diff --git a/chainladder/workflow/gridsearch.py b/chainladder/workflow/gridsearch.py index 01d807aa..9d498a5b 100644 --- a/chainladder/workflow/gridsearch.py +++ b/chainladder/workflow/gridsearch.py @@ -135,7 +135,7 @@ def predict(self, X, sample_weight=None): if sample_weight: fit_params = {} if not fit_params else fit_params fit_params[self.steps[-1][0] + '__sample_weight'] = sample_weight - return super().predict(X, y, **fit_params) + return super().predict(X, **fit_params) def to_json(self):