diff --git a/src/lerobot/datasets/lerobot_dataset.py b/src/lerobot/datasets/lerobot_dataset.py index ec419d55..9bbe07a5 100644 --- a/src/lerobot/datasets/lerobot_dataset.py +++ b/src/lerobot/datasets/lerobot_dataset.py @@ -708,7 +708,8 @@ class LeRobotDataset(torch.utils.data.Dataset): if not self._check_cached_episodes_sufficient(): raise FileNotFoundError("Cached dataset doesn't contain all requested episodes") except (AssertionError, FileNotFoundError, NotADirectoryError): - self.revision = get_safe_version(self.repo_id, self.revision) + if is_valid_version(self.revision): + self.revision = get_safe_version(self.repo_id, self.revision) self.download(download_videos) self.hf_dataset = self.load_hf_dataset()