Add simxarm back into tests

2024-03-25 16:35:46 +01:00
parent d3adaf1379
commit 058ac991eb
18 changed files with 44 additions and 19 deletions
--- a/lerobot/common/envs/simxarm/env.py
+++ b/lerobot/common/envs/simxarm/env.py
@@ -84,7 +84,7 @@ class SimxarmEnv(AbstractEnv):
        else:
            obs = {"state": torch.tensor(raw_obs["observation"], dtype=torch.float32)}

-        obs = TensorDict(obs, batch_size=[])
+        # obs = TensorDict(obs, batch_size=[])
        return obs

    def _reset(self, tensordict: Optional[TensorDict] = None):