This commit is contained in:
Pradeep Kadubandi 2025-01-15 01:50:38 -08:00 committed by Michel Aractingi
parent df7310ea40
commit 068efce3f8
1 changed files with 1 additions and 1 deletions

View File

@ -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: