Fix for the issue https://github.com/huggingface/lerobot/issues/638 (#639)

2025-01-15 01:50:38 -08:00
parent df7310ea40
commit 068efce3f8
1 changed files with 1 additions and 1 deletions
--- a/lerobot/scripts/control_robot.py
+++ b/lerobot/scripts/control_robot.py
@@ -309,7 +309,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)
            if reset_follower: