MilesCranmer commited on
Commit
4b48961
1 Parent(s): b271a60

Fix unit propagation test

Browse files
Files changed (1) hide show
  1. 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