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

Change card creation

This commit is contained in:
Simon Alibert 2024-10-20 14:01:10 +02:00
parent 9316cf46ef
commit e46bdb9d30
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lerobot/common/datasets/v2/convert_dataset_v1_to_v2.py
View File

@ -125,12 +125,12 @@ from lerobot.common.datasets.utils import (
DEFAULT_PARQUET_PATH, DEFAULT_PARQUET_PATH,
DEFAULT_VIDEO_PATH, DEFAULT_VIDEO_PATH,
create_branch, create_branch,
create_lerobot_dataset_card,
flatten_dict, flatten_dict,
get_hub_safe_version, get_hub_safe_version,
unflatten_dict, unflatten_dict,
) )
from lerobot.common.utils.utils import init_hydra_config from lerobot.common.utils.utils import init_hydra_config
from lerobot.scripts.push_dataset_to_hub import push_dataset_card_to_hub
V16 = "v1.6" V16 = "v1.6"
V20 = "v2.0" V20 = "v2.0"
@ -716,8 +716,9 @@ def convert_dataset(
revision=branch, revision=branch,
) )
card_text = f"[meta/info.json](meta/info.json)\n```json\n{json.dumps(metadata_v2_0, indent=4)}\n```" card = create_lerobot_dataset_card(tags=repo_tags, info=metadata_v2_0)
push_dataset_card_to_hub(repo_id=repo_id, revision=branch, tags=repo_tags, text=card_text) card.push_to_hub(repo_id=repo_id, repo_type="dataset", revision=branch)
if not test_branch: if not test_branch:
create_branch(repo_id=repo_id, branch=V20, repo_type="dataset") create_branch(repo_id=repo_id, branch=V20, repo_type="dataset")