remove internal rendering hooks

2024-03-20 09:23:23 +00:00
parent d16f6a93b3
commit b1ec3da035
5 changed files with 0 additions and 22 deletions
--- a/lerobot/common/envs/aloha/env.py
+++ b/lerobot/common/envs/aloha/env.py
@@ -164,7 +164,6 @@ class AlohaEnv(AbstractEnv):
            batch_size=[],
        )

-        self.call_rendering_hooks()
        return td

    def _step(self, tensordict: TensorDict):
@@ -189,8 +188,6 @@ class AlohaEnv(AbstractEnv):
                stacked_obs["state"] = torch.stack(list(self._prev_obs_state_queue))
            obs = stacked_obs

-        self.call_rendering_hooks()
-
        td = TensorDict(
            {
                "observation": TensorDict(obs, batch_size=[]),