MilesCranmer commited on
Commit
7763215
1 Parent(s): bb81a9a

test: fix early quit in complexity test

Browse files
Files changed (1) hide show
  1. pysr/test/test.py +1 -1
pysr/test/test.py CHANGED
@@ -184,7 +184,7 @@ class TestPipeline(unittest.TestCase):
184
  binary_operators=["*", "+"],
185
  verbosity=0,
186
  **self.default_test_kwargs,
187
- early_stop_condition="stop_if(l, c) = l < 1e-4 && c <= 7",
188
  )
189
  if case == 1:
190
  complexity_of_variables = [2, 3] + [
 
184
  binary_operators=["*", "+"],
185
  verbosity=0,
186
  **self.default_test_kwargs,
187
+ early_stop_condition=f"stop_if(l, c) = l < 1e-8 && c <= {7 if case == 1 else 5}",
188
  )
189
  if case == 1:
190
  complexity_of_variables = [2, 3] + [