From d2ef43436c69a4df1b10f4923467894410774bad Mon Sep 17 00:00:00 2001 From: Cadene Date: Sat, 23 Mar 2024 13:34:35 +0000 Subject: [PATCH] move from cadene to lerobot --- lerobot/common/datasets/abstract.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index 529bf6db..8295ed48 100644 --- a/lerobot/common/datasets/abstract.py +++ b/lerobot/common/datasets/abstract.py @@ -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.envs.transforms.transforms import Compose +HF_USER = "lerobot" + class AbstractExperienceReplay(TensorDictReplayBuffer): def __init__( @@ -106,7 +108,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer): if self.root is None: self.data_dir = Path( 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: