Update setup.py for imblearn

sarahshi · Dec 7, 2023 · 36504ba · 36504ba
1 parent ee14430
commit 36504ba
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/Train_Supervised_Variational.py b/Train_Supervised_Variational.py
@@ -26,10 +26,6 @@
 sys.path.append('src')
 import mineralML as mm
 
-import concurrent.futures
-from multiprocessing import freeze_support
-import itertools 
-
 from matplotlib import pyplot as plt
 from matplotlib import rc
 import seaborn as sns

diff --git a/environment.yml b/environment.yml
@@ -24,4 +24,5 @@ dependencies:
     - pip:
         - pyrolite
         - PyIRoGlass
-        - hdbscan
+        - hdbscan
+        - imblearn
diff --git a/none.pdf b/none.pdf
diff --git a/setup.py b/setup.py
@@ -34,7 +34,8 @@
             'scikit-learn',
             'scipy',
             'torch',
-            'hdbscan'
+            'hdbscan', 
+            'imblearn'
             ],
 
     classifiers=[