From bb5676ee5a8c97efc76c3acd3b813f6ff54a9aeb Mon Sep 17 00:00:00 2001
From: Francesco Capuano <francesco.capuano@huggingface.co>
Date: Thu, 6 Nov 2025 00:41:12 +0000
Subject: [PATCH] add: number of workers for merging datasets

---
 src/lerobot/scripts/lerobot_edit_dataset.py | 2 ++
 1 file changed, 2 insertions(+)

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}")