MilesCranmer commited on
Commit
a8f65d5
1 Parent(s): 2b8573e

Force version for scikit-learn requirement

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. setup.py +7 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  sympy
2
  pandas
3
  numpy
4
- scikit_learn
5
  julia
 
1
  sympy
2
  pandas
3
  numpy
4
+ scikit_learn >= 1.0.0
5
  julia
setup.py CHANGED
@@ -17,7 +17,13 @@ setuptools.setup(
17
  long_description=long_description,
18
  long_description_content_type="text/markdown",
19
  url="https://github.com/MilesCranmer/pysr",
20
- install_requires=["julia>=0.5.7", "numpy", "pandas", "sympy", "scikit-learn"],
 
 
 
 
 
 
21
  packages=setuptools.find_packages(),
22
  package_data={"pysr": ["../Project.toml", "../datasets/*"]},
23
  include_package_data=False,
 
17
  long_description=long_description,
18
  long_description_content_type="text/markdown",
19
  url="https://github.com/MilesCranmer/pysr",
20
+ install_requires=[
21
+ "julia>=0.5.7",
22
+ "numpy",
23
+ "pandas",
24
+ "sympy",
25
+ "scikit-learn >= 1.0.0",
26
+ ],
27
  packages=setuptools.find_packages(),
28
  package_data={"pysr": ["../Project.toml", "../datasets/*"]},
29
  include_package_data=False,