diff --git a/lerobot/common/datasets/aloha.py b/lerobot/common/datasets/aloha.py index ec58e701..e891ccdd 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 3ad6371f..a8a47da8 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 0f3f9d2f..4e1c1884 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 0fa5667c..844550e1 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 b00756e6..836b29cf 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