From 2a2b64889192b74129603f3f664c5f9e80654c18 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Tue, 2 Dec 2025 18:11:20 +0100
Subject: [PATCH] fix use local dir

---
 examples/dataset/slurm_port_egodex.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/dataset/slurm_port_egodex.py b/examples/dataset/slurm_port_egodex.py
index ec9c73e26..40d4d17e2 100644
--- a/examples/dataset/slurm_port_egodex.py
+++ b/examples/dataset/slurm_port_egodex.py
@@ -278,12 +278,14 @@ class PortEgoDexShards(PipelineStep):
 
         # 4. Create a LeRobot dataset for this shard
         shard_repo_id = f"{self.repo_id}_world_{world_size}_rank_{rank}"
+        shard_root = self.local_dir / shard_repo_id if self.local_dir else None
 
         dataset = LeRobotDataset.create(
             repo_id=shard_repo_id,
             fps=30,
             robot_type="hand",
             features=EGODEX_FEATURES,
+            root=shard_root,
         )
 
         # 5. Process each file