Merge pull request #39 from huggingface/user/rcadene/2024_03_20_updat…

…e_stats_v1.1 Update stats for Pusht and Aloha datasets (v1.0 -> v1.1)
huggingface · Mar 20, 2024 · 2b02210 · 2b02210
2 parents ec536ef + f23a53c
commit 2b02210
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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
diff --git a/tests/data/aloha_sim_transfer_cube_human/stats.pth b/tests/data/aloha_sim_transfer_cube_human/stats.pth
diff --git a/tests/data/aloha_sim_transfer_cube_scripted/stats.pth b/tests/data/aloha_sim_transfer_cube_scripted/stats.pth