MilesCranmer commited on
Commit
8fe6c62
1 Parent(s): 3fcdd9a

Turn back up complex test

Browse files
Files changed (1) hide show
  1. pysr/test/test.py +1 -1
pysr/test/test.py CHANGED
@@ -220,7 +220,7 @@ class TestPipeline(unittest.TestCase):
220
  model.fit(X, y)
221
  test_y = model.predict(X)
222
  self.assertTrue(np.issubdtype(test_y.dtype, np.complexfloating))
223
- self.assertLessEqual(np.average(np.abs(test_y - y) ** 2), 1e-3)
224
 
225
  def test_empty_operators_single_input_warm_start(self):
226
  X = self.rstate.randn(100, 1)
 
220
  model.fit(X, y)
221
  test_y = model.predict(X)
222
  self.assertTrue(np.issubdtype(test_y.dtype, np.complexfloating))
223
+ self.assertLessEqual(np.average(np.abs(test_y - y) ** 2), 1e-4)
224
 
225
  def test_empty_operators_single_input_warm_start(self):
226
  X = self.rstate.randn(100, 1)