Update setup.py

elyxlz · Feb 10, 2023 · fb5a0f3 · fb5a0f3
1 parent 4706822
commit fb5a0f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,7 @@
     'pynndescent',
     'llvmlite>=0.34.0',
     'torch>=1.6',
-    'sklearn',
+    'scikit-learn',
     'umap-learn',
     'pytorch_lightning',
     'dill',
@@ -34,4 +34,4 @@
     'License :: OSI Approved :: MIT License',
     'Programming Language :: Python :: 3.9',
   ],
-)
+)