MilesCranmer commited on
Commit
6e07d35
1 Parent(s): 8fe6c62

Needs longer for complex search

Browse files
Files changed (1) hide show
  1. pysr/test/test.py +1 -0
pysr/test/test.py CHANGED
@@ -217,6 +217,7 @@ class TestPipeline(unittest.TestCase):
217
  **self.default_test_kwargs,
218
  early_stop_condition="(loss, complexity) -> loss <= 1e-4 && complexity <= 6",
219
  )
 
220
  model.fit(X, y)
221
  test_y = model.predict(X)
222
  self.assertTrue(np.issubdtype(test_y.dtype, np.complexfloating))
 
217
  **self.default_test_kwargs,
218
  early_stop_condition="(loss, complexity) -> loss <= 1e-4 && complexity <= 6",
219
  )
220
+ model.niterations = DEFAULT_NITERATIONS * 10
221
  model.fit(X, y)
222
  test_y = model.predict(X)
223
  self.assertTrue(np.issubdtype(test_y.dtype, np.complexfloating))