Add new test_control_robot

2025-05-08 17:38:16 +02:00
parent f9db727647
commit 95f9b45418
3 changed files with 96 additions and 7 deletions
--- a/lerobot/record.py
+++ b/lerobot/record.py
@@ -195,7 +195,7 @@ def record_loop(


@draccus.wrap()
-def record(cfg: RecordConfig):
+def record(cfg: RecordConfig) -> LeRobotDataset:
    init_logging()
    logging.info(pformat(asdict(cfg)))
    if cfg.display_data: