Spaces:
Running
Running
Ensure equation state is reset for warm start test
Browse files
pysr/test/test_warm_start.py
CHANGED
|
@@ -70,6 +70,10 @@ class TestWarmStart(unittest.TestCase):
|
|
| 70 |
|
| 71 |
assert model.julia_state_ is not None
|
| 72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
model.warm_start = True
|
| 74 |
model.niterations = 0
|
| 75 |
model.max_evals = 0
|
|
|
|
| 70 |
|
| 71 |
assert model.julia_state_ is not None
|
| 72 |
|
| 73 |
+
# Reset saved equations; should be loaded from state!
|
| 74 |
+
model.equations_ = None
|
| 75 |
+
model.equation_file_contents_ = None
|
| 76 |
+
|
| 77 |
model.warm_start = True
|
| 78 |
model.niterations = 0
|
| 79 |
model.max_evals = 0
|