From 462e7469e84f6b4728f4682fb6a5723008538984 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Mon, 25 Mar 2024 12:27:26 +0100 Subject: [PATCH] Add xarm_lift_medium revision 1.0 to hub --- lerobot/common/datasets/abstract.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index 794d60a2..8d17cd78 100644 --- a/lerobot/common/datasets/abstract.py +++ b/lerobot/common/datasets/abstract.py @@ -21,7 +21,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer): def __init__( self, dataset_id: str, - version: str | None = None, + version: str | None = "v1.0", batch_size: int = None, *, shuffle: bool = True, @@ -45,11 +45,12 @@ class AbstractExperienceReplay(TensorDictReplayBuffer): ) # HACK: to remove before merge - self.data_dir = self.root / self.dataset_id - if not (self.data_dir / "replay_buffer").exists(): - storage = self._download_and_preproc_obsolete() - else: - storage = self._download_or_load_dataset() + # self.data_dir = self.root / self.dataset_id + # if not (self.data_dir / "replay_buffer").exists(): + # storage = self._download_and_preproc_obsolete() + # else: + # storage = self._download_or_load_dataset() + storage = self._download_or_load_dataset() super().__init__( storage=storage,