Merge pull request #39 from huggingface/user/rcadene/2024_03_20_update_stats_v1.1

Update stats for Pusht and Aloha datasets (v1.0 -> v1.1)
2024-03-20 18:41:15 +01:00
parent ec536ef0fa f23a53c3e4
commit 2b0221052a
5 changed files with 2 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.0",
+        version: str | None = "v1.1",
        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.0",
+        version: str | None = "v1.1",
        batch_size: int = None,
        *,
        shuffle: bool = True,
--- 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