[Spaces] Multidimensional observations reflected in spaces

proroklab · May 2, 2024 · 243d474 · 243d474
1 parent ffa6c60
commit 243d474
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vmas/simulator/environment/environment.py b/vmas/simulator/environment/environment.py
@@ -353,7 +353,7 @@ def get_agent_observation_space(self, agent: Agent, obs: AGENT_OBS_TYPE):
             return spaces.Box(
                 low=-np.float32("inf"),
                 high=np.float32("inf"),
-                shape=(len(obs[0]),),
+                shape=obs.shape[1:],
                 dtype=np.float32,
             )
         elif isinstance(obs, Dict):