Skip to content

Commit 243d474

Browse files
committed
[Spaces] Multidimensional observations reflected in spaces
1 parent ffa6c60 commit 243d474

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vmas/simulator/environment/environment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,7 @@ def get_agent_observation_space(self, agent: Agent, obs: AGENT_OBS_TYPE):
353353
return spaces.Box(
354354
low=-np.float32("inf"),
355355
high=np.float32("inf"),
356-
shape=(len(obs[0]),),
356+
shape=obs.shape[1:],
357357
dtype=np.float32,
358358
)
359359
elif isinstance(obs, Dict):

0 commit comments

Comments
 (0)