Tests cleaning & simplification (#81)

2024-04-18 14:47:42 +02:00
parent 0928afd37d
commit 7ad1909641
24 changed files with 277 additions and 157 deletions
--- a/lerobot/common/datasets/xarm.py
+++ b/lerobot/common/datasets/xarm.py
@@ -11,9 +11,8 @@ class XarmDataset(torch.utils.data.Dataset):
    https://huggingface.co/datasets/lerobot/xarm_lift_medium
    """

-    available_datasets = [
-        "xarm_lift_medium",
-    ]
+    # Copied from lerobot/__init__.py
+    available_datasets = ["xarm_lift_medium"]
    fps = 15
    image_keys = ["observation.image"]