diff --git a/lerobot/common/datasets/aloha.py b/lerobot/common/datasets/aloha.py index ec58e7010..e891ccdd7 100644 --- 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.0", + version: str | None = "v1.1", batch_size: int = None, *, shuffle: bool = True, diff --git a/lerobot/common/datasets/pusht.py b/lerobot/common/datasets/pusht.py index 3ad6371fb..a8a47da81 100644 --- 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.0", + version: str | None = "v1.1", batch_size: int = None, *, shuffle: bool = True, diff --git a/tests/data/aloha_sim_insertion_scripted/stats.pth b/tests/data/aloha_sim_insertion_scripted/stats.pth index 0f3f9d2f1..4e1c18844 100644 Binary files a/tests/data/aloha_sim_insertion_scripted/stats.pth and b/tests/data/aloha_sim_insertion_scripted/stats.pth differ diff --git a/tests/data/aloha_sim_transfer_cube_human/stats.pth b/tests/data/aloha_sim_transfer_cube_human/stats.pth index 0fa5667cf..844550e1e 100644 Binary files a/tests/data/aloha_sim_transfer_cube_human/stats.pth and b/tests/data/aloha_sim_transfer_cube_human/stats.pth differ diff --git a/tests/data/aloha_sim_transfer_cube_scripted/stats.pth b/tests/data/aloha_sim_transfer_cube_scripted/stats.pth index b00756e6c..836b29cf5 100644 Binary files a/tests/data/aloha_sim_transfer_cube_scripted/stats.pth and b/tests/data/aloha_sim_transfer_cube_scripted/stats.pth differ