nit in replay

huggingface · Dec 3, 2024 · 9cc4e26 · 9cc4e26
1 parent 75f5191
commit 9cc4e26
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/lerobot/scripts/control_sim_robot.py b/lerobot/scripts/control_sim_robot.py
@@ -246,7 +246,6 @@ def record(
             root=root,
             features=features,
             use_videos=video,
-            tolerance_s=1e-1,
             image_writer_processes=num_image_writer_processes,
             image_writer_threads=num_image_writer_threads_per_camera * num_cameras,
         )
@@ -495,7 +494,7 @@ def replay(
         help="Dataset identifier. By convention it should match '{hf_username}/{dataset_name}' (e.g. `lerobot/test`).",
     )
     parser_replay.add_argument(
-        "--episodes", nargs="+", type=int, default=[0], help="Indices of the episodes to replay."
+        "--episode", type=int, default=0, help="Index of the episodes to replay."
     )
 
     args = parser.parse_args()