diff --git a/src/dataset/diode_dataset.py b/src/dataset/diode_dataset.py
index 569d9a8..2699dea 100755
--- a/src/dataset/diode_dataset.py
+++ b/src/dataset/diode_dataset.py
@@ -28,7 +28,7 @@
 
 import numpy as np
 import torch
-from scipy import ndimage  # <--- Added for Sobel filter
+from scipy import ndimage
 
 from .base_depth_dataset import BaseDepthDataset, DepthFileNameMode, DatasetMode