Merge remote-tracking branch 'upstream/main' into user/alexander-soare/multistep_policy_and_serial_env

2024-03-20 09:01:45 +00:00
parent 52e149fbfd 2d5abbbd6f
commit d16f6a93b3
76 changed files with 92 additions and 14 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 = "v1.0",
        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,