Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_04_release_cleanup

2024-05-05 14:47:49 +02:00 · 2024-05-05 14:47:49 +02:00 · 17778fa3cb
parent dc56e9b930 6e6feae09e
commit 17778fa3cb
1 changed files with 1 additions and 1 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@ -338,7 +338,7 @@ def eval_policy(
                    target=write_video,
                    args=(
                        str(video_path),
-                        stacked_frames[: done_index + 2],  # + 2 to capture the observation frame after done
+                        stacked_frames[: done_index + 1],  # + 1 to capture the last observation
                        env.unwrapped.metadata["render_fps"],
                    ),
                )