From 253e495df237ccd7b6db1dfacac2cdeeba29bd82 Mon Sep 17 00:00:00 2001 From: Cadene Date: Tue, 9 Apr 2024 03:46:05 +0000 Subject: [PATCH] remove render(mode=visualization) --- lerobot/scripts/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index e7ba53fc..512bb451 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -86,10 +86,10 @@ def eval_policy( def maybe_render_frame(env): if save_video: # noqa: B023 if return_first_video: - visu = env.envs[0].render(mode="visualization") + visu = env.envs[0].render() visu = visu[None, ...] # add batch dim else: - visu = np.stack([env.render(mode="visualization") for env in env.envs]) + visu = np.stack([env.render() for env in env.envs]) ep_frames.append(visu) # noqa: B023 for _ in range(num_episodes):