From 380b836eee5f7cd61f023ef09dd5e0d9721d8c54 Mon Sep 17 00:00:00 2001 From: Pradeep Kadubandi Date: Wed, 15 Jan 2025 01:50:38 -0800 Subject: [PATCH] Fix for the issue https://github.com/huggingface/lerobot/issues/638 (#639) --- lerobot/scripts/control_robot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/control_robot.py b/lerobot/scripts/control_robot.py index 12eaf146..30f22cb0 100644 --- a/lerobot/scripts/control_robot.py +++ b/lerobot/scripts/control_robot.py @@ -300,7 +300,7 @@ def record( # TODO(rcadene): add an option to enable teleoperation during reset # Skip reset for the last episode to be recorded if not events["stop_recording"] and ( - (dataset.num_episodes < num_episodes - 1) or events["rerecord_episode"] + (recorded_episodes < num_episodes - 1) or events["rerecord_episode"] ): log_say("Reset the environment", play_sounds) reset_environment(robot, events, reset_time_s)