Add replay_buffer directory in pusht datasets + aloha (WIP)

2024-03-19 15:49:45 +00:00
parent 099a465367
commit 6a1a29386a
20 changed files with 53 additions and 8 deletions
--- a/lerobot/common/datasets/aloha.py
+++ b/lerobot/common/datasets/aloha.py
@@ -84,6 +84,7 @@ class AlohaExperienceReplay(AbstractExperienceReplay):
    def __init__(
        self,
        dataset_id: str,
+        version: str | None = None,
        batch_size: int = None,
        *,
        shuffle: bool = True,
@@ -99,6 +100,7 @@ class AlohaExperienceReplay(AbstractExperienceReplay):

        super().__init__(
            dataset_id,
+            version,
            batch_size,
            shuffle=shuffle,
            root=root,