From e1ac5dc62fac32b6d452a45019870e602fbd884c Mon Sep 17 00:00:00 2001 From: Cadene Date: Sun, 7 Apr 2024 17:20:54 +0000 Subject: [PATCH] fix aloha pixels env test --- lerobot/common/envs/aloha/env.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/lerobot/common/envs/aloha/env.py b/lerobot/common/envs/aloha/env.py index 22cd0116..bd14e6d8 100644 --- a/lerobot/common/envs/aloha/env.py +++ b/lerobot/common/envs/aloha/env.py @@ -48,8 +48,15 @@ class AlohaEnv(gym.Env): dtype=np.float64, ) elif self.obs_type == "pixels": - self.observation_space = spaces.Box( - low=0, high=255, shape=(self.observation_height, self.observation_width, 3), dtype=np.uint8 + self.observation_space = spaces.Dict( + { + "top": spaces.Box( + low=0, + high=255, + shape=(self.observation_height, self.observation_width, 3), + dtype=np.uint8, + ) + } ) elif self.obs_type == "pixels_agent_pos": self.observation_space = spaces.Dict(