Done when sim_step >= sim_max_time instead of >

LucasAlegre · Feb 22, 2023 · cf46b7d · cf46b7d
1 parent 53c5611
commit cf46b7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sumo_rl/environment/env.py b/sumo_rl/environment/env.py
@@ -299,7 +299,7 @@ def _apply_actions(self, actions):
 
     def _compute_dones(self):
         dones = {ts_id: False for ts_id in self.ts_ids}
-        dones['__all__'] = self.sim_step > self.sim_max_time
+        dones['__all__'] = self.sim_step >= self.sim_max_time
         return dones
 
     def _compute_info(self):