MilesCranmer commited on
Commit
e56007b
·
1 Parent(s): 56cb021

Correct use of Xresampled

Browse files
Files changed (1) hide show
  1. test/test.py +1 -2
test/test.py CHANGED
@@ -217,11 +217,10 @@ class TestPipeline(unittest.TestCase):
217
  unary_operators=["cos"],
218
  select_k_features=3,
219
  early_stop_condition=1e-4, # Stop once most accurate equation is <1e-4 MSE
220
- Xresampled=Xresampled,
221
  maxsize=12,
222
  **self.default_test_kwargs,
223
  )
224
- model.fit(X, y)
225
  model.set_params(model_selection="accuracy")
226
  model.predict(X)
227
  self.assertLess(np.average((model.predict(X) - y) ** 2), 1e-4)
 
217
  unary_operators=["cos"],
218
  select_k_features=3,
219
  early_stop_condition=1e-4, # Stop once most accurate equation is <1e-4 MSE
 
220
  maxsize=12,
221
  **self.default_test_kwargs,
222
  )
223
+ model.fit(X, y, Xresampled=Xresampled)
224
  model.set_params(model_selection="accuracy")
225
  model.predict(X)
226
  self.assertLess(np.average((model.predict(X) - y) ** 2), 1e-4)