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

Add copied from tags

This commit is contained in:
Simon Alibert 2024-04-18 11:10:14 +02:00
parent 673dd8fd3f
commit 088b807cf2
4 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CONTRIBUTING.md
View File

@ -75,7 +75,7 @@ and follow the same api design.
When implementing a new dataset class (e.g. `AlohaDataset`) follow these steps: When implementing a new dataset class (e.g. `AlohaDataset`) follow these steps:
- Update `available_datasets` in `lerobot/__init__.py` - Update `available_datasets` in `lerobot/__init__.py`
- Set the required `available_datasets` class attribute using the previously updated `lerobot.available_datasets` - Copy it in the required `available_datasets` class attribute
When implementing a new environment (e.g. `gym_aloha`), follow these steps: When implementing a new environment (e.g. `gym_aloha`), follow these steps:
- Update `available_envs`, `available_tasks_per_env` and `available_datasets` in `lerobot/__init__.py` - Update `available_envs`, `available_tasks_per_env` and `available_datasets` in `lerobot/__init__.py`

1
lerobot/common/datasets/aloha.py
View File

@ -14,6 +14,7 @@ class AlohaDataset(torch.utils.data.Dataset):
https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_scripted https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_scripted
""" """
# Copied from lerobot/__init__.py
available_datasets = [ available_datasets = [
"aloha_sim_insertion_human", "aloha_sim_insertion_human",
"aloha_sim_insertion_scripted", "aloha_sim_insertion_scripted",

1
lerobot/common/datasets/pusht.py
View File

@ -17,6 +17,7 @@ class PushtDataset(torch.utils.data.Dataset):
If `None`, no shift is applied to current timestamp and the data from the current frame is loaded. If `None`, no shift is applied to current timestamp and the data from the current frame is loaded.
""" """
# Copied from lerobot/__init__.py
available_datasets = ["pusht"] available_datasets = ["pusht"]
fps = 10 fps = 10
image_keys = ["observation.image"] image_keys = ["observation.image"]

1
lerobot/common/datasets/xarm.py
View File

@ -11,6 +11,7 @@ class XarmDataset(torch.utils.data.Dataset):
https://huggingface.co/datasets/lerobot/xarm_lift_medium https://huggingface.co/datasets/lerobot/xarm_lift_medium
""" """
# Copied from lerobot/__init__.py
available_datasets = ["xarm_lift_medium"] available_datasets = ["xarm_lift_medium"]
fps = 15 fps = 15
image_keys = ["observation.image"] image_keys = ["observation.image"]