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

Add safe_disconnect to replay

This commit is contained in:
Remi Cadene 2024-10-15 19:38:40 +02:00
parent 19d410a372
commit 29f6abc997
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/scripts/control_robot.py
View File

@ -310,6 +310,7 @@ def record(
return lerobot_dataset return lerobot_dataset
@safe_disconnect
def replay( def replay(
robot: Robot, episode: int, fps: int | None = None, root="data", repo_id="lerobot/debug", play_sounds=True robot: Robot, episode: int, fps: int | None = None, root="data", repo_id="lerobot/debug", play_sounds=True
): ):