Fix record

This commit is contained in:
Remi Cadene 2024-10-24 15:27:33 +02:00
parent 102e1d4436
commit abbf9c5bee
1 changed files with 3 additions and 1 deletions

View File

@ -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,