diff --git a/lerobot/scripts/control_robot.py b/lerobot/scripts/control_robot.py index 3fdef1ff..425247e6 100644 --- a/lerobot/scripts/control_robot.py +++ b/lerobot/scripts/control_robot.py @@ -310,6 +310,7 @@ def record( return lerobot_dataset +@safe_disconnect def replay( robot: Robot, episode: int, fps: int | None = None, root="data", repo_id="lerobot/debug", play_sounds=True ):