fix: update scikit-learn dependency

2022-12-05 16:31:33 +01:00
parent 0649d5bb45
commit 7b23983887
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@ INSTALL_REQUIRES = [
    "torch>=1.3.1",
    "torchvision>=0.7.4",
    "numpy>=1.9.1",
-    "sklearn",
+    "scikit-learn",
    "matplotlib",
 ]
 DATASETS = [