Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
6e07d35
1
Parent(s):
8fe6c62
Needs longer for complex search
Browse files- 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))
|