fixing a bug with scikit learn install

mims-harvard · Apr 22, 2024 · 4cd0850 · 4cd0850
1 parent 19c1e7a
commit 4cd0850
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='spectrae',
-    version='0.1.1',    
+    version='1.0.1',    
     description='SPECTRA: The spectral framework for model evaluation',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',
@@ -14,7 +14,7 @@
     install_requires=['networkx',
                       'numpy', 
                       'torch', 
-                      'sklearn',
+                      'scikit-learn',
                       'pandas',
                       'tqdm',
                       'cell-gears',