Remove additional frame from eval render (#137)

2024-05-05 12:49:53 +01:00 · 2024-05-05 12:49:53 +01:00 · 6e6feae09e
commit 6e6feae09e
parent f6fe162432
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"],
                    ),
                )