Merge remote-tracking branch 'upstream/main' into finish_examples

2024-04-01 11:31:31 +01:00
parent b7c9c33072 2a98cc71ed
commit f1148b8c2d
8 changed files with 406 additions and 43 deletions
--- a/lerobot/common/envs/aloha/env.py
+++ b/lerobot/common/envs/aloha/env.py
@@ -206,7 +206,7 @@ class AlohaEnv(AbstractEnv):
        if self.from_pixels:
            if isinstance(self.image_size, int):
                image_shape = (3, self.image_size, self.image_size)
-            elif OmegaConf.is_list(self.image_size):
+            elif OmegaConf.is_list(self.image_size) or isinstance(self.image_size, list):
                assert len(self.image_size) == 3  # c h w
                assert self.image_size[0] == 3  # c is RGB
                image_shape = tuple(self.image_size)