Add system info to pettingzoo api

LucasAlegre · Jul 23, 2023 · 13102e2 · 13102e2
1 parent 5443f6c
commit 13102e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sumo_rl/environment/env.py b/sumo_rl/environment/env.py
@@ -539,7 +539,7 @@ def compute_info(self):
         infos = self.env._compute_info()
         for a in self.agents:
             for k, v in infos.items():
-                if k.startswith(a):
+                if k.startswith(a) or k.startswith("system"):
                     self.infos[a][k] = v
 
     def observation_space(self, agent):