Issue #134: Reset latest_step_duration

AVSLab · May 9, 2024 · d9398b8 · d9398b8
1 parent f4ae2c1
commit d9398b8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/bsk_rl/envs/general_satellite_tasking/gym_env.py b/src/bsk_rl/envs/general_satellite_tasking/gym_env.py
@@ -120,7 +120,7 @@ def __init__(
         self.failure_penalty = failure_penalty
         self.time_limit = time_limit
         self.terminate_on_time_limit = terminate_on_time_limit
-        self.latest_step_duration = 0
+        self.latest_step_duration = 0.0
         self.render_mode = render_mode
 
     def _minimum_env_model(self) -> type[EnvironmentModel]:
@@ -205,6 +205,8 @@ def reset(
         np.random.seed(self.seed)
         self._generate_env_args()
 
+        self.latest_step_duration = 0.0
+
         self.env_features.reset()
         self.data_manager.reset()