Change type of reward in EnvStep (#2274)

rlworkgroup · May 9, 2021 · 1e7ca02 · 1e7ca02
1 parent 5bc155b
commit 1e7ca02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/garage/_environment.py b/src/garage/_environment.py
@@ -111,7 +111,7 @@ class EnvStep:
 
     env_spec: EnvSpec
     action: np.ndarray
-    reward: np.ndarray
+    reward: float
     observation: np.ndarray
     env_info: Dict[str, np.ndarray or dict]
     step_type: StepType