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

Update load_tasks doc

This commit is contained in:
Simon Alibert 2024-10-15 11:06:28 +02:00
parent 835ab5a81b
commit da78bbfd16
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -180,15 +180,7 @@ def load_info(repo_id: str, version: str, local_dir: Path) -> dict:
def load_tasks(repo_id: str, version: str, local_dir: Path) -> dict: def load_tasks(repo_id: str, version: str, local_dir: Path) -> dict:
"""tasks contains all the tasks of the dataset, indexed by their task_index. """tasks contains all the tasks of the dataset, indexed by their task_index."""
Example:
```json
{
"0": "Pick the Lego block and drop it in the box on the right."
}
```
"""
fpath = hf_hub_download( fpath = hf_hub_download(
repo_id, filename="meta/tasks.json", local_dir=local_dir, repo_type="dataset", revision=version repo_id, filename="meta/tasks.json", local_dir=local_dir, repo_type="dataset", revision=version
) )