fix: set traci default view if it doesn't exist

LucasAlegre · Mar 2, 2024 · a49f3da · a49f3da
1 parent a134ebb
commit a49f3da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sumo_rl/environment/env.py b/sumo_rl/environment/env.py
@@ -236,6 +236,8 @@ def _start_simulation(self):
             self.sumo = traci.getConnection(self.label)
 
         if self.use_gui or self.render_mode is not None:
+            if "DEFAULT_VIEW" not in dir(traci.gui):
+                traci.gui.DEFAULT_VIEW = 'View #0'
             self.sumo.gui.setSchema(traci.gui.DEFAULT_VIEW, "real world")
 
     def reset(self, seed: Optional[int] = None, **kwargs):