Fix record

huggingface · Oct 24, 2024 · abbf9c5 · abbf9c5
1 parent 102e1d4
commit abbf9c5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lerobot/scripts/control_robot.py b/lerobot/scripts/control_robot.py
@@ -207,6 +207,8 @@ def record(
     num_image_writer_threads_per_camera: int = 4,
     display_cameras: bool = True,
     play_sounds: bool = True,
+    tags: str = None,
+    force_override: bool = False,
 ) -> LeRobotDataset:
     # TODO(rcadene): Add option to record logs
     listener = None
@@ -237,8 +239,8 @@ def record(
     dataset = LeRobotDataset.create(
         repo_id,
         fps,
-        robot,
         root=root,
+        robot=robot,
         image_writer_processes=num_image_writer_processes,
         image_writer_threads_per_camera=num_image_writer_threads_per_camera,
         use_videos=video,