wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/control_robot.py
View File

@ -309,7 +309,7 @@ def record(
# TODO(rcadene): add an option to enable teleoperation during reset # TODO(rcadene): add an option to enable teleoperation during reset
# Skip reset for the last episode to be recorded # Skip reset for the last episode to be recorded
if not events["stop_recording"] and ( 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) log_say("Reset the environment", play_sounds)
if reset_follower: if reset_follower: