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