Force version for scikit-learn requirement

MilesCranmer · Jun 22, 2022 · a8f65d5 · a8f65d5
1 parent 2b8573e
commit a8f65d5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 sympy
 pandas
 numpy
-scikit_learn
+scikit_learn >= 1.0.0
 julia
diff --git a/setup.py b/setup.py
@@ -17,7 +17,13 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/MilesCranmer/pysr",
-    install_requires=["julia>=0.5.7", "numpy", "pandas", "sympy", "scikit-learn"],
+    install_requires=[
+        "julia>=0.5.7",
+        "numpy",
+        "pandas",
+        "sympy",
+        "scikit-learn >= 1.0.0",
+    ],
     packages=setuptools.find_packages(),
     package_data={"pysr": ["../Project.toml", "../datasets/*"]},
     include_package_data=False,