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

feat(warning): adding a warning when depth unit is not specified in the dataset

Browse Source
This commit is contained in:
CarolinePascal
2026-06-30 22:16:40 +02:00
parent bcc71bf73b
commit afa189fc72
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/dataset_metadata.py
+10
View File
@@ -367,6 +367,16 @@ class LeRobotDatasetMetadata:
``output_unit`` on read. This converts the unit-bearing stat entries so ``output_unit`` on read. This converts the unit-bearing stat entries so
stats match the frames consumers see. stats match the frames consumers see.
""" """
missing_unit_keys = [
key for key in self.depth_keys if (self.features[key].get("info") or {}).get("depth_unit") is None
]
if missing_unit_keys:
logging.warning(
f"Depth feature(s) {missing_unit_keys} have no recorded 'depth_unit' in their info. "
f"Depth maps and stats for these keys will be returned AS IS, with no unit conversion "
f"to the requested output unit {output_unit!r}. Re-record the dataset or set 'depth_unit' "
f"in the feature info (meta/info.json) to enable conversion."
)
if self.stats is None: if self.stats is None:
return return
for key in self.depth_keys: for key in self.depth_keys: