hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Uploaded droid 1.0.1

Browse Source
This commit is contained in:
Remi Cadene
2025-05-08 15:14:15 +00:00
parent 0309a9fcbc
commit 1ecaeabad0
3 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/datasets/utils.py
View File

@@ -126,9 +126,8 @@ def load_nested_dataset(pq_dir: Path) -> Dataset:
raise FileNotFoundError(f"Provided directory does not contain any parquet file: {pq_dir}")
# TODO(rcadene): set num_proc to accelerate conversion to pyarrow
return concatenate_datasets(
[Dataset.from_parquet(str(path)) for path in sorted(pq_dir.glob("*/*.parquet"))]
)
datasets = [Dataset.from_parquet(str(path)) for path in paths]
return concatenate_datasets(datasets)
def get_parquet_num_frames(parquet_path):