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

Add tag

This commit is contained in:
Simon Alibert 2025-02-19 15:01:44 +01:00
parent 2487228ea7
commit 6fe42a72db
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/port_datasets/pusht_zarr.py
View File

@ -3,9 +3,10 @@ from pathlib import Path
import numpy as np import numpy as np
import torch import torch
from huggingface_hub import HfApi
from lerobot.common.constants import HF_LEROBOT_HOME from lerobot.common.constants import HF_LEROBOT_HOME
from lerobot.common.datasets.lerobot_dataset import LeRobotDataset from lerobot.common.datasets.lerobot_dataset import CODEBASE_VERSION, LeRobotDataset
from lerobot.common.datasets.push_dataset_to_hub._download_raw import download_raw from lerobot.common.datasets.push_dataset_to_hub._download_raw import download_raw
PUSHT_TASK = "Push the T-shaped blue block onto the T-shaped green target surface." PUSHT_TASK = "Push the T-shaped blue block onto the T-shaped green target surface."
@ -203,6 +204,8 @@ def main(raw_dir: Path, repo_id: str, mode: str = "video", push_to_hub: bool = T
if push_to_hub: if push_to_hub:
dataset.push_to_hub() dataset.push_to_hub()
hub_api = HfApi()
hub_api.create_tag(repo_id, tag=CODEBASE_VERSION, repo_type="dataset")
if __name__ == "__main__": if __name__ == "__main__":