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__( def __init__(
self, self,
dataset_id: str, dataset_id: str,
version: str | None = None, version: str | None = "v1.0",
batch_size: int = None, batch_size: int = None,
*, *,
shuffle: bool = True, shuffle: bool = True,
@ -45,11 +45,12 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
) )
# HACK: to remove before merge # HACK: to remove before merge
self.data_dir = self.root / self.dataset_id # self.data_dir = self.root / self.dataset_id
if not (self.data_dir / "replay_buffer").exists(): # if not (self.data_dir / "replay_buffer").exists():
storage = self._download_and_preproc_obsolete() # storage = self._download_and_preproc_obsolete()
else: # else:
storage = self._download_or_load_dataset() # storage = self._download_or_load_dataset()
storage = self._download_or_load_dataset()
super().__init__( super().__init__(
storage=storage, storage=storage,