diff --git a/src/lerobot/datasets/dataset_reader.py b/src/lerobot/datasets/dataset_reader.py index dd68b6156..ff053feed 100644 --- a/src/lerobot/datasets/dataset_reader.py +++ b/src/lerobot/datasets/dataset_reader.py @@ -326,6 +326,8 @@ class DatasetReader: if self._image_transforms is not None: image_keys = self._meta.camera_keys for cam in image_keys: + if cam not in self._meta.depth_keys: + continue item[cam] = self._image_transforms(item[cam]) # Add task as a string