diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index f5b87b34..5b69a13a 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -75,7 +75,7 @@ and follow the same api design. When implementing a new dataset class (e.g. `AlohaDataset`) follow these steps: - 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: - Update `available_envs`, `available_tasks_per_env` and `available_datasets` in `lerobot/__init__.py` diff --git a/lerobot/common/datasets/aloha.py b/lerobot/common/datasets/aloha.py index 87ee57a8..4769a2bf 100644 --- a/lerobot/common/datasets/aloha.py +++ b/lerobot/common/datasets/aloha.py @@ -14,6 +14,7 @@ class AlohaDataset(torch.utils.data.Dataset): https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_scripted """ + # Copied from lerobot/__init__.py available_datasets = [ "aloha_sim_insertion_human", "aloha_sim_insertion_scripted", diff --git a/lerobot/common/datasets/pusht.py b/lerobot/common/datasets/pusht.py index b9d06ba4..c5c06bf6 100644 --- a/lerobot/common/datasets/pusht.py +++ b/lerobot/common/datasets/pusht.py @@ -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. """ + # Copied from lerobot/__init__.py available_datasets = ["pusht"] fps = 10 image_keys = ["observation.image"] diff --git a/lerobot/common/datasets/xarm.py b/lerobot/common/datasets/xarm.py index 6cbef535..711ff642 100644 --- a/lerobot/common/datasets/xarm.py +++ b/lerobot/common/datasets/xarm.py @@ -11,6 +11,7 @@ class XarmDataset(torch.utils.data.Dataset): https://huggingface.co/datasets/lerobot/xarm_lift_medium """ + # Copied from lerobot/__init__.py available_datasets = ["xarm_lift_medium"] fps = 15 image_keys = ["observation.image"]