tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Enhance dataset cards

Browse Source
This commit is contained in:
Simon Alibert
2024-11-20 11:57:27 +01:00
parent f43e5d07f5
commit c6ad495176
6 changed files with 265 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/datasets/lerobot_dataset.py
View File

@@ -467,10 +467,10 @@ class LeRobotDataset(torch.utils.data.Dataset):
def push_to_hub(
self,
tags: list | None = None,
text: str | None = None,
license: str | None = "apache-2.0",
push_videos: bool = True,
private: bool = False,
**card_kwargs,
) -> None:
if not self.consolidated:
raise RuntimeError(
@@ -495,7 +495,9 @@ class LeRobotDataset(torch.utils.data.Dataset):
repo_type="dataset",
ignore_patterns=ignore_patterns,
)
card = create_lerobot_dataset_card(tags=tags, text=text, info=self.meta.info, license=license)
card = create_lerobot_dataset_card(
tags=tags, dataset_info=self.meta.info, license=license, **card_kwargs
)
card.push_to_hub(repo_id=self.repo_id, repo_type="dataset")
create_branch(repo_id=self.repo_id, branch=CODEBASE_VERSION, repo_type="dataset")