Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
4b48961
1
Parent(s):
b271a60
Fix unit propagation test
Browse files- pysr/test/test.py +1 -1
pysr/test/test.py
CHANGED
@@ -1129,8 +1129,8 @@ class TestDimensionalConstraints(unittest.TestCase):
|
|
1129 |
|
1130 |
# Try warm start, but with no units provided (should
|
1131 |
# be a different dataset, and thus different result):
|
1132 |
-
model.fit(X, y)
|
1133 |
model.early_stop_condition = "(l, c) -> l < 1e-6 && c == 1"
|
|
|
1134 |
self.assertEqual(model.equations_.iloc[0].complexity, 1)
|
1135 |
self.assertLess(model.equations_.iloc[0].loss, 1e-6)
|
1136 |
|
|
|
1129 |
|
1130 |
# Try warm start, but with no units provided (should
|
1131 |
# be a different dataset, and thus different result):
|
|
|
1132 |
model.early_stop_condition = "(l, c) -> l < 1e-6 && c == 1"
|
1133 |
+
model.fit(X, y)
|
1134 |
self.assertEqual(model.equations_.iloc[0].complexity, 1)
|
1135 |
self.assertLess(model.equations_.iloc[0].loss, 1e-6)
|
1136 |
|