Skip to content

Commit 3570db7

Browse files
PradeepKadubandiJIy3AHKO
authored andcommitted
Fix for the issue huggingface#638 (huggingface#639)
1 parent d4f60f6 commit 3570db7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lerobot/scripts/control_robot.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,7 @@ def record(
300300
# TODO(rcadene): add an option to enable teleoperation during reset
301301
# Skip reset for the last episode to be recorded
302302
if not events["stop_recording"] and (
303-
(dataset.num_episodes < num_episodes - 1) or events["rerecord_episode"]
303+
(recorded_episodes < num_episodes - 1) or events["rerecord_episode"]
304304
):
305305
log_say("Reset the environment", play_sounds)
306306
reset_environment(robot, events, reset_time_s)

0 commit comments

Comments
 (0)