diff --git a/src/lerobot/scripts/lerobot_edit_dataset.py b/src/lerobot/scripts/lerobot_edit_dataset.py index 83ba027bc..36ae35d2e 100644 --- a/src/lerobot/scripts/lerobot_edit_dataset.py +++ b/src/lerobot/scripts/lerobot_edit_dataset.py @@ -103,6 +103,7 @@ class SplitConfig: class MergeConfig: type: str = "merge" repo_ids: list[str] | None = None + num_workers: int | None = None @dataclass @@ -215,6 +216,7 @@ def handle_merge(cfg: EditDatasetConfig) -> None: datasets, output_repo_id=cfg.repo_id, output_dir=output_dir, + num_workers=cfg.operation.num_workers, ) logging.info(f"Merged dataset saved to {output_dir}")