From 212a5ab29bbc12886657ad0e1adf2f386d4e9a3a Mon Sep 17 00:00:00 2001
From: Marina Barannikov <marina.barannikov@huggingface.co>
Date: Fri, 31 May 2024 16:28:08 +0000
Subject: [PATCH] Updated implementation on MultiLeRobotDataset

---
 lerobot/common/datasets/lerobot_dataset.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py
index 491ba00bd..7da5f4301 100644
--- a/lerobot/common/datasets/lerobot_dataset.py
+++ b/lerobot/common/datasets/lerobot_dataset.py
@@ -382,10 +382,6 @@ class MultiLeRobotDataset(torch.utils.data.Dataset):
             if data_key in item:
                 del item[data_key]
 
-        if self.transform is not None:
-            for cam in self.camera_keys:
-                item[cam] = self.transform(item[cam])
-
         return item
 
     def __repr__(self):