Add safe_disconnect to replay

2024-10-15 19:38:40 +02:00
parent 19d410a372
commit 29f6abc997
1 changed files with 1 additions and 0 deletions
--- 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
 ):