hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

fix(recording): re-recording episode doesn't increase count of recording episodes (#1395)

Browse Source
This commit is contained in:
Steven Palma
2025-06-27 16:02:51 +02:00
committed by GitHub
parent 2f9ba4e2cc
commit 69901b9b6a
2 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lerobot/record.py
View File

@@ -286,7 +286,8 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
listener, events = init_keyboard_listener()
for recorded_episodes in range(cfg.dataset.num_episodes):
recorded_episodes = 0
while recorded_episodes < cfg.dataset.num_episodes and not events["stop_recording"]:
log_say(f"Recording episode {dataset.num_episodes}", cfg.play_sounds)
record_loop(
robot=robot,
@@ -324,9 +325,7 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
continue
dataset.save_episode()
if events["stop_recording"]:
break
recorded_episodes += 1
log_say("Stop recording", cfg.play_sounds, blocking=True)