diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py index 491ba00b..7da5f430 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):