ready for review

2024-03-21 10:18:50 +00:00
parent d323993569
commit acf1174447
12 changed files with 282 additions and 85 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -50,6 +50,7 @@ def eval_policy(

        def maybe_render_frame(env: EnvBase, _):
            if save_video or (return_first_video and i == 0):  # noqa: B023
+                # TODO now: generalize kwarg or maybe just remove it
                ep_frames.append(env.render())  # noqa: B023

        with torch.inference_mode():