[QOL] Enable teleoperation during environment reset (#725)

2025-02-25 19:28:26 +01:00
parent 65db5afe1c
commit 8699a28be0
2 changed files with 9 additions and 16 deletions
--- a/lerobot/scripts/control_robot.py
+++ b/lerobot/scripts/control_robot.py
@@ -299,7 +299,7 @@ def record(
            (recorded_episodes < cfg.num_episodes - 1) or events["rerecord_episode"]
        ):
            log_say("Reset the environment", cfg.play_sounds)
-            reset_environment(robot, events, cfg.reset_time_s)
+            reset_environment(robot, events, cfg.reset_time_s, cfg.fps)

        if events["rerecord_episode"]:
            log_say("Re-record episode", cfg.play_sounds)