admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(datasets): skip warning when MultiLeRobotDataset features are identical (#3019)

Browse Source 
Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
Michio Sun
2026-02-27 01:42:22 +09:00
committed by GitHub
parent fde9d08281
commit 4e54be1334
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
@@ -1771,11 +1771,12 @@ class MultiLeRobotDataset(torch.utils.data.Dataset):
) )
for repo_id, ds in zip(self.repo_ids, self._datasets, strict=True): for repo_id, ds in zip(self.repo_ids, self._datasets, strict=True):
extra_keys = set(ds.features).difference(intersection_features) extra_keys = set(ds.features).difference(intersection_features)
logging.warning( if extra_keys:
f"keys {extra_keys} of {repo_id} were disabled as they are not contained in all the " logging.warning(
"other datasets." f"keys {extra_keys} of {repo_id} were disabled as they are not contained in all the "
) "other datasets."
self.disabled_features.update(extra_keys) )
self.disabled_features.update(extra_keys)
self.image_transforms = image_transforms self.image_transforms = image_transforms
self.delta_timestamps = delta_timestamps self.delta_timestamps = delta_timestamps