fix use local dir

2026-05-15 16:49:55 +00:00 · 2025-12-02 18:11:20 +01:00
parent cf36f4b873
commit 2a2b648891
1 changed files with 2 additions and 0 deletions
@@ -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