From 9e8bc0729d51dac1dfdc7914423bb8b50982d86c Mon Sep 17 00:00:00 2001
From: CarolinePascal <caroline8.pascal@gmail.com>
Date: Thu, 25 Jun 2026 11:33:07 +0200
Subject: [PATCH] fix(image transforms): excluding depth frames from images
 transforms

---
 src/lerobot/datasets/dataset_reader.py | 2 ++
 1 file changed, 2 insertions(+)

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