MilesCranmer commited on
Commit
32c64f7
1 Parent(s): 7ae2181

Fix version of pyjulia

Browse files
Files changed (2) hide show
  1. setup.py +1 -1
  2. test/test.py +1 -1
setup.py CHANGED
@@ -15,7 +15,7 @@ setuptools.setup(
15
  long_description=long_description,
16
  long_description_content_type="text/markdown",
17
  url="https://github.com/MilesCranmer/pysr",
18
- install_requires=["julia", "numpy", "pandas", "sympy", "scikit-learn"],
19
  packages=setuptools.find_packages(),
20
  package_data={"pysr": ["../Project.toml", "../datasets/*"]},
21
  include_package_data=False,
 
15
  long_description=long_description,
16
  long_description_content_type="text/markdown",
17
  url="https://github.com/MilesCranmer/pysr",
18
+ install_requires=["julia>=0.5.7", "numpy", "pandas", "sympy", "scikit-learn"],
19
  packages=setuptools.find_packages(),
20
  package_data={"pysr": ["../Project.toml", "../datasets/*"]},
21
  include_package_data=False,
test/test.py CHANGED
@@ -95,7 +95,7 @@ class TestPipeline(unittest.TestCase):
95
  np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
96
 
97
  # Test if repeated fit works:
98
- regressor.set_params(ncyclesperiteration=50)
99
  regressor.fit(X, y)
100
 
101
  self.assertLessEqual(regressor.equations.iloc[-1]["loss"], 1e-4)
 
95
  np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
96
 
97
  # Test if repeated fit works:
98
+ regressor.set_params(niterations=0)
99
  regressor.fit(X, y)
100
 
101
  self.assertLessEqual(regressor.equations.iloc[-1]["loss"], 1e-4)