MilesCranmer commited on
Commit
7ae2181
1 Parent(s): 7153ae1

Fix state saving test

Browse files
Files changed (1) hide show
  1. test/test.py +1 -1
test/test.py CHANGED
@@ -95,7 +95,7 @@ class TestPipeline(unittest.TestCase):
95
  np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
96
 
97
  # Test if repeated fit works:
98
- regressor.set_params(ncyclesperiteration=1)
99
  regressor.fit(X, y)
100
 
101
  self.assertLessEqual(regressor.equations.iloc[-1]["loss"], 1e-4)
 
95
  np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
96
 
97
  # Test if repeated fit works:
98
+ regressor.set_params(ncyclesperiteration=50)
99
  regressor.fit(X, y)
100
 
101
  self.assertLessEqual(regressor.equations.iloc[-1]["loss"], 1e-4)