wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add xarm_lift_medium revision 1.0 to hub

This commit is contained in:
Simon Alibert 2024-03-25 12:27:26 +01:00
parent 298d391b26
commit 462e7469e8
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lerobot/common/datasets/abstract.py
View File

@ -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,