Add copied from tags

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`

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",

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"]

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"]