[Spaces] Multidimensional observations reflected in spaces

matteobettini · matteobettini · commit 243d4740d064 · 2024-05-02T10:15:40.000+01:00
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):

Original file line number	Diff line number	Diff line change
`@@ -353,7 +353,7 @@ def get_agent_observation_space(self, agent: Agent, obs: AGENT_OBS_TYPE):`
`353`	`353`	`return spaces.Box(`
`354`	`354`	`low=-np.float32("inf"),`
`355`	`355`	`high=np.float32("inf"),`
`356`		`- shape=(len(obs[0]),),`
	`356`	`+ shape=obs.shape[1:],`
`357`	`357`	`dtype=np.float32,`
`358`	`358`	`)`
`359`	`359`	`elif isinstance(obs, Dict):`