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