Ran pre-commit run --all-files

2024-02-29 13:37:48 +01:00
parent 0e0e760e43
commit 7e024fdce6
16 changed files with 124 additions and 237 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -37,10 +37,11 @@ def eval_policy(
        tensordict = env.reset()

        ep_frames = []
+
        if save_video or (return_first_video and i == 0):

            def rendering_callback(env, td=None):
-                ep_frames.append(env.render())
+                ep_frames.append(env.render())  # noqa: B023

            # render first frame before rollout
            rendering_callback(env)