From bc424f6fb79c6420adfc3b4c0d340fba1c8026df Mon Sep 17 00:00:00 2001
From: Alexander Soare <alexander.soare159@gmail.com>
Date: Sun, 5 May 2024 12:02:33 +0100
Subject: [PATCH] Remove addtional frame from eval render

---
 lerobot/scripts/eval.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py
index e3afac41..363362bd 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"],
                     ),
                 )