Add scikit-learn to requirements (for least-squares equalization)

FrancescoConti · May 8, 2020 · a1c6fdb · a1c6fdb
1 parent 5e692ff
commit a1c6fdb
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 2 deletions.
diff --git a/.github/requirements/1.3.1.txt b/.github/requirements/1.3.1.txt
@@ -3,3 +3,4 @@ torchvision>=0.4.1,<0.5.0
 numpy
 tqdm
 packaging
+scikit-learn
diff --git a/.github/requirements/1.4.0.txt b/.github/requirements/1.4.0.txt
@@ -3,3 +3,4 @@ torchvision>=0.5.0,<0.6.0
 numpy
 tqdm
 packaging
+scikit-learn
diff --git a/.github/requirements/1.5.0.txt b/.github/requirements/1.5.0.txt
@@ -3,3 +3,4 @@ torchvision>=0.6.0,<0.7.0
 numpy
 tqdm
 packaging
+scikit-learn
diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,4 @@ torchvision>=0.4.1
 numpy
 tqdm
 packaging
+scikit-learn
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="pytorch-nemo",
-    version="0.0.4",
+    version="0.0.5",
     author="Francesco Conti",
     author_email="[email protected]",
     description="NEural Minimizer for pytOrch",
@@ -24,6 +24,7 @@
         "torchvision>=0.4.1",
         "numpy",
         "tqdm",
-        "packaging"
+        "packaging",
+        "scikit-learn"
     ]
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,3 +3,4 @@ torchvision>=0.4.1,<0.5.0 @@
     numpy
     tqdm
     packaging
+    scikit-learn