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

Merge 6d0e058186 into 78fd1a1e04

This commit is contained in:
Ben Sprenger 2025-03-28 13:15:29 +08:00 committed by GitHub
parent 78fd1a1e04 6d0e058186
commit aa0bcaffea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,7 +27,6 @@ import torch
import torch.utils import torch.utils
from datasets import concatenate_datasets, load_dataset from datasets import concatenate_datasets, load_dataset
from huggingface_hub import HfApi, snapshot_download from huggingface_hub import HfApi, snapshot_download
from huggingface_hub.constants import REPOCARD_NAME
from huggingface_hub.errors import RevisionNotFoundError from huggingface_hub.errors import RevisionNotFoundError
from lerobot.common.constants import HF_LEROBOT_HOME from lerobot.common.constants import HF_LEROBOT_HOME
@ -560,11 +559,10 @@ class LeRobotDataset(torch.utils.data.Dataset):
else: else:
hub_api.upload_folder(**upload_kwargs) hub_api.upload_folder(**upload_kwargs)
if not hub_api.file_exists(self.repo_id, REPOCARD_NAME, repo_type="dataset", revision=branch): card = create_lerobot_dataset_card(
card = create_lerobot_dataset_card( tags=tags, dataset_info=self.meta.info, license=license, **card_kwargs
tags=tags, dataset_info=self.meta.info, license=license, **card_kwargs )
) card.push_to_hub(repo_id=self.repo_id, repo_type="dataset", revision=branch)
card.push_to_hub(repo_id=self.repo_id, repo_type="dataset", revision=branch)
if tag_version: if tag_version:
with contextlib.suppress(RevisionNotFoundError): with contextlib.suppress(RevisionNotFoundError):