From 7a2b9daf5e7be3d1c1e7716b338e58cc0fc2cc76 Mon Sep 17 00:00:00 2001 From: nlahaye Date: Tue, 26 Apr 2022 22:11:21 -0700 Subject: [PATCH] Allowing for models without built-in scoring to be used --- dask_ml/wrappers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: