admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

chores(warning): improving warnings and errors for episodes filtering

Browse Source
This commit is contained in:
CarolinePascal
2026-05-07 18:28:54 +02:00
parent dd920bf9e3
commit 5e963b2815
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/lerobot_dataset.py
+6 -5
View File
@@ -204,7 +204,6 @@ class LeRobotDataset(torch.utils.data.Dataset):
self.reader = None self.reader = None
self.set_image_transforms(image_transforms) self.set_image_transforms(image_transforms)
self.delta_timestamps = delta_timestamps self.delta_timestamps = delta_timestamps
self.episodes = episodes
self.tolerance_s = tolerance_s self.tolerance_s = tolerance_s
self.revision = revision if revision else CODEBASE_VERSION self.revision = revision if revision else CODEBASE_VERSION
self._video_backend = video_backend if video_backend else get_safe_default_codec() self._video_backend = video_backend if video_backend else get_safe_default_codec()
@@ -223,14 +222,16 @@ class LeRobotDataset(torch.utils.data.Dataset):
self.root = self.meta.root self.root = self.meta.root
self.revision = self.meta.revision self.revision = self.meta.revision
if episodes is not None and any(episode>=self.meta.total_episodes or episode<0 for episode in episodes):
logger.warning(f"Some episodes in the provided episodes list are out of range for this dataset ({self.meta.total_episodes}).")
if episode_filter is not None: if episode_filter is not None:
resolved = self.meta.filter_episodes(episode_filter, candidates=episodes) resolved = self.meta.filter_episodes(episode_filter, candidates=episodes)
pool = len(episodes) if episodes is not None else self.meta.total_episodes
if not resolved: if not resolved:
raise ValueError(f"episode_filter did not match any episode over {pool} episode(s).") raise ValueError(f"The episode filter did not match any episode. Make sure the filter and episodes list are valid and compatible.")
logger.info(f"episode_filter matched {len(resolved)} episode(s) over {pool} episode(s).") logger.info(f"The episode filter matched {len(resolved)} episode(s).")
episodes = resolved episodes = resolved
self.episodes = resolved self.episodes = episodes
# Create reader (hf_dataset loaded below) # Create reader (hf_dataset loaded below)
self.reader = DatasetReader( self.reader = DatasetReader(