wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix safe_version

This commit is contained in:
Simon Alibert 2024-10-14 13:51:40 +02:00
parent cbc51e1341
commit f96773de10
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
convert_dataset_v1_to_v2.py
View File

@ -393,8 +393,8 @@ def convert_dataset(
tasks_col: Path | None = None, tasks_col: Path | None = None,
robot_config: dict | None = None, robot_config: dict | None = None,
): ):
v1 = get_hub_safe_version(repo_id, V16) v1 = get_hub_safe_version(repo_id, V16, enforce_v2=False)
v1x_dir = local_dir / v1 / repo_id v1x_dir = local_dir / V16 / repo_id
v20_dir = local_dir / V20 / repo_id v20_dir = local_dir / V20 / repo_id
v1x_dir.mkdir(parents=True, exist_ok=True) v1x_dir.mkdir(parents=True, exist_ok=True)
v20_dir.mkdir(parents=True, exist_ok=True) v20_dir.mkdir(parents=True, exist_ok=True)
@ -493,10 +493,10 @@ def convert_dataset(
convert_stats_to_json(v1x_dir / "meta_data", v20_dir / "meta") convert_stats_to_json(v1x_dir / "meta_data", v20_dir / "meta")
#### TODO: delete #### TODO: delete
# repo_id = f"aliberts/{repo_id.split('/')[1]}" repo_id = f"aliberts/{repo_id.split('/')[1]}"
# if hub_api.repo_exists(repo_id=repo_id, repo_type="dataset"): # if hub_api.repo_exists(repo_id=repo_id, repo_type="dataset"):
# hub_api.delete_repo(repo_id=repo_id, repo_type="dataset") # hub_api.delete_repo(repo_id=repo_id, repo_type="dataset")
# hub_api.create_repo(repo_id=repo_id, repo_type="dataset", exist_ok=True) hub_api.create_repo(repo_id=repo_id, repo_type="dataset", exist_ok=True)
#### ####
with contextlib.suppress(EntryNotFoundError): with contextlib.suppress(EntryNotFoundError):