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

add typing hints

Browse Source
This commit is contained in:
Michel Aractingi
2025-11-06 09:12:09 +01:00
parent b5dcd70d2c
commit ca5a4a7ae5
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/v30/convert_dataset_v21_to_v30.py
+3 -5
View File
@@ -175,13 +175,11 @@ def convert_tasks(root, new_root):
write_tasks(df_tasks, new_root) write_tasks(df_tasks, new_root)
def concat_data_files(paths_to_cat, new_root, chunk_idx, file_idx, image_keys): def concat_data_files(
paths_to_cat: list[Path], new_root: Path, chunk_idx: int, file_idx: int, image_keys: list[str]
):
"""Concatenate multiple parquet data files into a single file. """Concatenate multiple parquet data files into a single file.
This function uses HuggingFace datasets to properly handle image embedding,
ensuring the output has the same internal structure as datasets created through
live recording. This is critical for training performance.
Args: Args:
paths_to_cat: List of parquet file paths to concatenate paths_to_cat: List of parquet file paths to concatenate
new_root: Root directory for the new dataset new_root: Root directory for the new dataset