diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 7b63f964..2a61aaea 100644 --- 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"], ), )