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

move from cadene to lerobot

This commit is contained in:
Cadene 2024-03-23 13:34:35 +00:00
parent 40f3783fca
commit d2ef43436c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,6 +14,8 @@ from torchrl.data.replay_buffers.storages import TensorStorage, _collate_id
from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer
from torchrl.envs.transforms.transforms import Compose from torchrl.envs.transforms.transforms import Compose
HF_USER = "lerobot"
class AbstractExperienceReplay(TensorDictReplayBuffer): class AbstractExperienceReplay(TensorDictReplayBuffer):
def __init__( def __init__(
@ -106,7 +108,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
if self.root is None: if self.root is None:
self.data_dir = Path( self.data_dir = Path(
snapshot_download( snapshot_download(
repo_id=f"cadene/{self.dataset_id}", repo_type="dataset", revision=self.version repo_id=f"{HF_USER}/{self.dataset_id}", repo_type="dataset", revision=self.version
) )
) )
else: else: