Add render_mode attribute to PZ env

LucasAlegre · May 28, 2024 · 2a8b6e5 · 2a8b6e5
1 parent f2d1d87
commit 2a8b6e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sumo_rl/environment/env.py b/sumo_rl/environment/env.py
@@ -517,6 +517,7 @@ def __init__(self, **kwargs):
 
         self.seed()
         self.env = SumoEnvironment(**self._kwargs)
+        self.render_mode = self.env.render_mode
 
         self.agents = self.env.ts_ids
         self.possible_agents = self.env.ts_ids