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

Update lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py

Browse Source 
Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Thomas Wolf
2024-05-29 15:30:21 +02:00
committed by GitHub
parent e2f690e779
commit 785db44bd5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py
View File

@@ -45,7 +45,8 @@ def load_from_raw(raw_dir: Path, out_dir: Path):
reference_files = list(raw_dir.glob("observation.images.cam_*.parquet"))
if len(reference_files) == 0:
raise ValueError(f"Missing reference files for camera, starting with in '{raw_dir}'")
reference_key = reference_files[0].stem
# select first camera in alphanumeric order
reference_key = sorted(reference_files)[0].stem
reference_df = pd.read_parquet(raw_dir / f"{reference_key}.parquet")
reference_df = reference_df[["timestamp_utc", reference_key]]