v1.2

2024-03-23 11:41:56 +00:00 · 2024-03-23 11:41:56 +00:00 · 40f3783fca
parent a80d9c0257
commit 40f3783fca
8 changed files with 5 additions and 2 deletions
--- a/lerobot/common/datasets/aloha.py
+++ b/lerobot/common/datasets/aloha.py
@ -84,7 +84,7 @@ class AlohaExperienceReplay(AbstractExperienceReplay):
    def __init__(
        self,
        dataset_id: str,
-        version: str | None = "v1.1",
+        version: str | None = "v1.2",
        batch_size: int = None,
        *,
        shuffle: bool = True,
--- a/lerobot/common/datasets/pusht.py
+++ b/lerobot/common/datasets/pusht.py
@ -87,7 +87,7 @@ class PushtExperienceReplay(AbstractExperienceReplay):
    def __init__(
        self,
        dataset_id: str,
-        version: str | None = "v1.1",
+        version: str | None = "v1.2",
        batch_size: int = None,
        *,
        shuffle: bool = True,
--- a/tests/data/aloha_sim_insertion_human/stats.pth
+++ b/tests/data/aloha_sim_insertion_human/stats.pth
--- a/tests/data/aloha_sim_insertion_scripted/stats.pth
+++ b/tests/data/aloha_sim_insertion_scripted/stats.pth
--- a/tests/data/aloha_sim_transfer_cube_human/stats.pth
+++ b/tests/data/aloha_sim_transfer_cube_human/stats.pth
--- a/tests/data/aloha_sim_transfer_cube_scripted/stats.pth
+++ b/tests/data/aloha_sim_transfer_cube_scripted/stats.pth
--- a/tests/data/pusht/stats.pth
+++ b/tests/data/pusht/stats.pth
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@ -22,6 +22,9 @@ from .utils import DEVICE, init_config
        ("simxarm", "diffusion", []),
        ("pusht", "diffusion", []),
        ("aloha", "act", ["env.task=sim_insertion_scripted"]),
+        ("aloha", "act", ["env.task=sim_insertion_human"]),
+        ("aloha", "act", ["env.task=sim_transfer_cube_scripted"]),
+        ("aloha", "act", ["env.task=sim_transfer_cube_human"]),
    ],
 )
 def test_concrete_policy(env_name, policy_name, extra_overrides):