From 6e6feae09e4792d09a2c286c909b3d78898ca3c2 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Sun, 5 May 2024 12:49:53 +0100 Subject: [PATCH] Remove additional frame from eval render (#137) --- 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 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"], ), )