From a1c6fdbd338545606de16c0e3d2f36de1b8070be Mon Sep 17 00:00:00 2001 From: Francesco Conti Date: Wed, 6 May 2020 09:09:16 +0200 Subject: [PATCH] Add scikit-learn to requirements (for least-squares equalization) --- .github/requirements/1.3.1.txt | 1 + .github/requirements/1.4.0.txt | 1 + .github/requirements/1.5.0.txt | 1 + requirements.txt | 1 + setup.py | 5 +++-- 5 files changed, 7 insertions(+), 2 deletions(-) diff --git a/.github/requirements/1.3.1.txt b/.github/requirements/1.3.1.txt index 9488ba8..790f97f 100644 --- 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 index a949fb2..97c021d 100644 --- 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 index ea5d6fe..7719af3 100644 --- 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 index 25312c6..5ff121a 100644 --- 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 index 1a388db..157566b 100644 --- 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="f.conti@unibo.it", description="NEural Minimizer for pytOrch", @@ -24,6 +24,7 @@ "torchvision>=0.4.1", "numpy", "tqdm", - "packaging" + "packaging", + "scikit-learn" ] )