Aloha done

2024-03-19 16:03:42 +00:00
parent 6a1a29386a
commit 10034e85c4
57 changed files with 1 additions and 1 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 = None,
+        version: str | None = "v1.0",
        batch_size: int = None,
        *,
        shuffle: bool = True,
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/action.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/action.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/episode.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/episode.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/frame_id.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/frame_id.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/done.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/done.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/image/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/next/observation/state.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/observation/image/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/observation/meta.json
--- a/tests/data/aloha_sim_insertion_human/replay_buffer/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_human/replay_buffer/observation/state.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/action.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/action.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/episode.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/episode.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/frame_id.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/frame_id.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/done.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/done.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/state.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/meta.json
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/state.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/action.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/action.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/episode.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/episode.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/frame_id.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/frame_id.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/done.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/done.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/image/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/image/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/image/top.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/image/top.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/state.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/next/observation/state.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/image/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/image/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/image/top.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/image/top.memmap
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/meta.json
--- a/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/state.memmap
+++ b/tests/data/aloha_sim_transfer_cube_human/replay_buffer/observation/state.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/action.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/action.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/episode.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/episode.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/frame_id.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/frame_id.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/done.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/done.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/image/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/image/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/image/top.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/image/top.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/state.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/next/observation/state.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/image/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/image/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/image/top.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/image/top.memmap
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/meta.json
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/meta.json
--- a/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/state.memmap
+++ b/tests/data/aloha_sim_transfer_cube_scripted/replay_buffer/observation/state.memmap