Update setup.py

Co-authored-by: Roman Yurchak <rth.yurchak@gmail.com>
snorkel-team · Feb 12, 2021 · 8b5db26 · 8b5db26
1 parent 65ffc6b
commit 8b5db26
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -37,7 +37,8 @@ before_install:
   - sudo update-java-alternatives -s java-1.8.0-openjdk-amd64
 
 install:
-  - pip install -U pip setuptools wheel
+  - pip install -U pip wheel
+  - pip install -U pip setuptools
   - pip install -U tox==3.12.0
   - pip install -U codecov
   - tox --notest

diff --git a/requirements.txt b/requirements.txt
@@ -17,7 +17,7 @@ pandas>=0.25.0,<2.0.0
 tqdm>=4.33.0,<5.0.0
 
 # Internal models
-scikit-learn>=0.23.0,<0.25.0
+scikit-learn>=0.20.2,<0.25.0
 torch>=1.2.0,<2.0.0
 munkres>=1.0.6
 
@@ -33,9 +33,7 @@ tensorboard>=1.14.0,<2.0.0
 spacy>=2.1.0,<3.0.0
 
 # Scale-up/out
-dask-glm==0.2.0
-dask-ml==1.7.0
-dask[array,dataframe]==2.18.0
+dask[complete]>=2.1.0,<2.30.0
 
 # Serialization
 dill>=0.3.0,<0.4.0

diff --git a/setup.py b/setup.py
@@ -40,7 +40,7 @@
         "scipy>=1.2.0,<2.0.0",
         "pandas>=0.25.0,<2.0.0",
         "tqdm>=4.33.0,<5.0.0",
-        "scikit-learn>=0.23.0,<0.25.0",
+        "scikit-learn>=0.20.2,<0.25.0",
         "torch>=1.2.0,<2.0.0",
         "tensorboard>=1.14.0,<2.0.0",
         "networkx>=2.2,<2.4",