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/simxarm.py
+++ b/lerobot/common/datasets/simxarm.py
@@ -40,6 +40,7 @@ class SimxarmExperienceReplay(AbstractExperienceReplay):
    def __init__(
        self,
        dataset_id: str,
+        version: str | None = None,
        batch_size: int = None,
        *,
        shuffle: bool = True,
@@ -53,6 +54,7 @@ class SimxarmExperienceReplay(AbstractExperienceReplay):
    ):
        super().__init__(
            dataset_id,
+            version,
            batch_size,
            shuffle=shuffle,
            root=root,