MilesCranmer commited on
Commit
98ccd5e
1 Parent(s): c526fa1

Add `turbo` parameter with test

Browse files
Files changed (2) hide show
  1. pysr/sr.py +0 -1
  2. test/test.py +2 -1
pysr/sr.py CHANGED
@@ -1587,7 +1587,6 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1587
  early_stop_condition=early_stop_condition,
1588
  seed=seed,
1589
  deterministic=self.deterministic,
1590
- extend_user_operators=False,
1591
  define_helper_functions=False,
1592
  )
1593
 
 
1587
  early_stop_condition=early_stop_condition,
1588
  seed=seed,
1589
  deterministic=self.deterministic,
 
1590
  define_helper_functions=False,
1591
  )
1592
 
test/test.py CHANGED
@@ -70,12 +70,13 @@ class TestPipeline(unittest.TestCase):
70
  print(model.equations_)
71
  self.assertLessEqual(model.get_best()["loss"], 1e-4)
72
 
73
- def test_multiprocessing(self):
74
  y = self.X[:, 0]
75
  model = PySRRegressor(
76
  **self.default_test_kwargs,
77
  procs=2,
78
  multithreading=False,
 
79
  early_stop_condition="stop_if(loss, complexity) = loss < 1e-4 && complexity == 1",
80
  )
81
  model.fit(self.X, y)
 
70
  print(model.equations_)
71
  self.assertLessEqual(model.get_best()["loss"], 1e-4)
72
 
73
+ def test_multiprocessing_turbo(self):
74
  y = self.X[:, 0]
75
  model = PySRRegressor(
76
  **self.default_test_kwargs,
77
  procs=2,
78
  multithreading=False,
79
+ turbo=True,
80
  early_stop_condition="stop_if(loss, complexity) = loss < 1e-4 && complexity == 1",
81
  )
82
  model.fit(self.X, y)