diff --git a/datasets/augmentations.py b/datasets/augmentations.py
index 850dcea..e047676 100644
--- a/datasets/augmentations.py
+++ b/datasets/augmentations.py
@@ -11,9 +11,9 @@
 import numpy as np
 from detectron2.config import CfgNode
 from fvcore.transforms.transform import Transform
+from scipy.ndimage import gaussian_filter
 
 sys.path.append(str(Path(__file__).resolve().parent.joinpath("..")))
-from scipy.ndimage import gaussian_filter
 
 from datasets.transforms import (
     AffineTransform,
diff --git a/environment.yml b/environment.yml
index 4a2c638..b13e426 100644
--- a/environment.yml
+++ b/environment.yml
@@ -22,6 +22,7 @@ dependencies:
   - timm
   - gunicorn
   - pillow
+  - shapely
   # - pygments #Optional for colors
   - pip:
       - natsort
diff --git a/environment_dev.yml b/environment_dev.yml
index 1118b7c..70f7fa7 100644
--- a/environment_dev.yml
+++ b/environment_dev.yml
@@ -23,6 +23,7 @@ dependencies:
   - timm
   - gunicorn
   - pillow
+  - shapely
   # - pygments #Optional for colors
   - pip:
       - natsort