dask / dask-ml

Scalable Machine Learning with Dask
http://ml.dask.org
BSD 3-Clause "New" or "Revised" License
903 stars 256 forks source link

Compatibility with scikit-learn 1.6 #1002

Open flying-sheep opened 6 days ago

flying-sheep commented 6 days ago

Describe the issue: dask_ml.model_selection imports the private function _safe_tags from the private module sklearn.utils._tags.

That no longer works in scikit-learn 1.6.0rc1

Minimal Complete Verifiable Example:

import dask_ml.model_selection

Anything else we need to know?: no

Environment: irrelevant

TomAugspurger commented 6 days ago

Looks like estimator._get_tags might be an option? https://scikit-learn.org/stable/developers/develop.html#estimator-tags