From 502fdc06302d27c0f108f3280ae4e90769304a7a Mon Sep 17 00:00:00 2001 From: Antoine Date: Mon, 20 Oct 2025 18:45:09 +0200 Subject: [PATCH] fix dataset revision (#2260) Co-authored-by: Steven Palma --- src/lerobot/datasets/lerobot_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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()