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/pusht.py
+++ b/lerobot/common/datasets/pusht.py
@@ -87,6 +87,7 @@ class PushtExperienceReplay(AbstractExperienceReplay):
    def __init__(
        self,
        dataset_id: str,
+        version: str | None = "v1.0",
        batch_size: int = None,
        *,
        shuffle: bool = True,
@@ -100,6 +101,7 @@ class PushtExperienceReplay(AbstractExperienceReplay):
    ):
        super().__init__(
            dataset_id,
+            version,
            batch_size,
            shuffle=shuffle,
            root=root,