ready for review

2024-04-08 13:10:19 +01:00
parent 1bab4a1dd5
commit 863f28ffd8
7 changed files with 92 additions and 168 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -86,7 +86,9 @@ def eval_policy(
    def maybe_render_frame(env):
        if save_video:  # noqa: B023
            if return_first_video:
-                visu = env.envs[0].render(mode="visualization")
+                # TODO(now): Put mode back in.
+                visu = env.envs[0].render()
+                # visu = env.envs[0].render(mode="visualization")
                visu = visu[None, ...]  # add batch dim
            else:
                # TODO(now): Put mode back in.