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
diff --git a/requirements.txt b/requirements.txt
index c582702..b3f2d66 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,4 @@ joblib
 osfclient
 ipywidgets
 ramp-workflow==0.6.0
-
+imbalanced-learn