MilesCranmer commited on
Commit
9137841
1 Parent(s): 857a9ad

Ensure large size test quits early even with fail

Browse files
Files changed (1) hide show
  1. test/test.py +1 -1
test/test.py CHANGED
@@ -322,7 +322,7 @@ class TestMiscellaneous(unittest.TestCase):
322
 
323
  def test_size_warning(self):
324
  """Ensure that a warning is given for a large input size."""
325
- model = PySRRegressor()
326
  X = np.random.randn(10001, 2)
327
  y = np.random.randn(10001)
328
  with warnings.catch_warnings():
 
322
 
323
  def test_size_warning(self):
324
  """Ensure that a warning is given for a large input size."""
325
+ model = PySRRegressor(max_evals=10000, populations=2)
326
  X = np.random.randn(10001, 2)
327
  y = np.random.randn(10001)
328
  with warnings.catch_warnings():