From 34454748f407fe4c87d0939de459f0a3adc906a1 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Mon, 9 Mar 2026 15:06:55 +0100 Subject: [PATCH] fix(datasets) --- .../datasets/v2/convert_dataset_v1_to_v2.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/src/lerobot/datasets/v2/convert_dataset_v1_to_v2.py b/src/lerobot/datasets/v2/convert_dataset_v1_to_v2.py index 034f4fbdc..d2032d4d8 100644 --- a/src/lerobot/datasets/v2/convert_dataset_v1_to_v2.py +++ b/src/lerobot/datasets/v2/convert_dataset_v1_to_v2.py @@ -120,6 +120,7 @@ from huggingface_hub import HfApi from huggingface_hub.errors import EntryNotFoundError, HfHubHTTPError from safetensors.torch import load_file +from lerobot.datasets.backward_compatibility import CompatibilityError from lerobot.datasets.utils import ( DEFAULT_CHUNK_SIZE, DEFAULT_PARQUET_PATH, @@ -131,6 +132,7 @@ from lerobot.datasets.utils import ( create_branch, create_lerobot_dataset_card, flatten_dict, + get_repo_versions, get_safe_version, load_json, unflatten_dict, @@ -455,8 +457,16 @@ def convert_dataset( test_branch: str | None = None, **card_kwargs, ): - v1 = get_safe_version(repo_id, V16) - v1x_dir = local_dir / V16 / repo_id + try: + v1 = get_safe_version(repo_id, V16) + except CompatibilityError: + hub_versions = get_repo_versions(repo_id) + v1x_versions = [v for v in hub_versions if v.major == 1] + if not v1x_versions: + raise + v1 = f"v{max(v1x_versions)}" + logging.warning(f"v1.6 not found for {repo_id}, falling back to {v1}") + v1x_dir = local_dir / v1 / repo_id v20_dir = local_dir / V20 / repo_id v1x_dir.mkdir(parents=True, exist_ok=True) v20_dir.mkdir(parents=True, exist_ok=True)