diff --git a/environment.yml b/environment.yml
index 0ad71d4..5e363fc 100644
--- a/environment.yml
+++ b/environment.yml
@@ -18,7 +18,8 @@ dependencies:
   - tensorflow-cpu
   - pytorch-cpu
   - torchvision-cpu
-  
+  - imbalanced-learn
+
   - pip
   - pip:
     - ramp-workflow==0.6.0