diff --git a/dask_ml/wrappers.py b/dask_ml/wrappers.py index 02edc1d22..2b78062ac 100644 --- a/dask_ml/wrappers.py +++ b/dask_ml/wrappers.py @@ -556,7 +556,7 @@ def _postfit_estimator(self): return self.estimator_ def _fit_for_estimator(self, estimator, X, y, **fit_kwargs): - check_scoring(estimator, self.scoring) + check_scoring(estimator, self.scoring, allow_none=True) if not dask.is_dask_collection(X) and not dask.is_dask_collection(y): result = estimator.partial_fit(X=X, y=y, **fit_kwargs) else: