Merge pull request #2059 from ngam/fix

fixing sklearn dep

Author: Balandat

.conda/meta.yaml

@@ -18,7 +18,7 @@ requirements:
   run:
     - pytorch>=1.10
     - numpy
-    - scikit-learn<=1.0.2  # newer versions require py3.8
+    - scikit-learn
     - scipy
 
 test:

.github/workflows/run_test_suite.yml

@@ -70,6 +70,7 @@ jobs:
     - name: Install dependencies
       run: |
         pip install pytest nbval jupyter tqdm matplotlib torchvision scipy
+        pip install -r requirements.txt
         python setup.py build develop
         pip install "pyro-ppl<=1.8";
     - name: Run example notebooks

environment.yml

@@ -4,5 +4,5 @@ channels:
 dependencies:
   - pytorch
   - numpy
-  - scikit-learn<=1.0.2  # newer versions require py3.8
+  - scikit-learn
   - scipy

requirements.txt

@@ -1,4 +1,4 @@
 torch>=1.10
 numpy
-scikit-learn<=1.0.2  # newer versions require py3.8
+scikit-learn
 scipy

setup.py

@@ -35,7 +35,7 @@ def find_version(*file_paths):
         install_requires = []
 except ImportError:
     pass
-install_requires += ["numpy", "scikit-learn<=1.0.2", "scipy"]
+install_requires += ["numpy", "scikit-learn", "scipy"]
 
 
 # Run the setup