From 785db44bd5dc984c88f83b53de15fbd9b3cddde1 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Wed, 29 May 2024 15:30:21 +0200 Subject: [PATCH] Update lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py Co-authored-by: Remi --- .../common/datasets/push_dataset_to_hub/aloha_dora_format.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py index c2e1497e..2b4b31cf 100644 --- a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py +++ b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py @@ -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]]