From b752833f3f4e8b12981e29c902a852b1533c4046 Mon Sep 17 00:00:00 2001 From: Cadene Date: Fri, 15 Mar 2024 13:19:18 +0000 Subject: [PATCH] fix download --- lerobot/common/datasets/abstract.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index e9613310..34b33c2e 100644 --- a/lerobot/common/datasets/abstract.py +++ b/lerobot/common/datasets/abstract.py @@ -96,7 +96,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer): def _download_or_load_dataset(self) -> torch.StorageBase: if self.root is None: - self.data_dir = snapshot_download(repo_id=f"cadene/{self.dataset_id}", repo_type="dataset") + self.data_dir = Path(snapshot_download(repo_id=f"cadene/{self.dataset_id}", repo_type="dataset")) else: self.data_dir = self.root / self.dataset_id return TensorStorage(TensorDict.load_memmap(self.data_dir))