admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 12:09:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix use local dir

Browse Source
This commit is contained in:
Pepijn
2025-12-02 18:11:20 +01:00
parent cf36f4b873
commit 2a2b648891
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/dataset/slurm_port_egodex.py
+2
View File
@@ -278,12 +278,14 @@ class PortEgoDexShards(PipelineStep):
# 4. Create a LeRobot dataset for this shard # 4. Create a LeRobot dataset for this shard
shard_repo_id = f"{self.repo_id}_world_{world_size}_rank_{rank}" 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( dataset = LeRobotDataset.create(
repo_id=shard_repo_id, repo_id=shard_repo_id,
fps=30, fps=30,
robot_type="hand", robot_type="hand",
features=EGODEX_FEATURES, features=EGODEX_FEATURES,
root=shard_root,
) )
# 5. Process each file # 5. Process each file