Spaces:
Running
Running
Turn back up complex test
Browse files- 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-
|
| 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)
|