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/pusht/env.py
+++ b/lerobot/common/envs/pusht/env.py
@@ -116,7 +116,6 @@ class PushtEnv(AbstractEnv):
            batch_size=[],
        )

-        self.call_rendering_hooks()
        return td

    def _step(self, tensordict: TensorDict):
@@ -139,8 +138,6 @@ class PushtEnv(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=[]),