Skip to content

Fix sklearn-dev build#450

Merged
TomAugspurger merged 2 commits intodask:masterfrom
jrbourbeau:fix-sklearn-dev
Jan 31, 2019
Merged

Fix sklearn-dev build#450
TomAugspurger merged 2 commits intodask:masterfrom
jrbourbeau:fix-sklearn-dev

Conversation

@jrbourbeau
Copy link
Copy Markdown
Member

The r2_score metric in scikit-learn was recently updated (see scikit-learn/scikit-learn#12855) to raise a warning when passed single sample. This is causing our sklearn-dev build to fail.

This PR updates test_search_cv_results_none_param in tests/model_selection/dask_searchcv/test_model_selection_sklearn.py to not pass a single sample to r2_score.

@jrbourbeau jrbourbeau changed the title Decrease n_splits Fix sklearn-dev build Jan 31, 2019
@jrbourbeau
Copy link
Copy Markdown
Member Author

Looks like the test failures here are related to the new pandas 0.24 release. I suspect it has to do with SparseArrays now implementing the ExtensionArray interface. Any thoughts @TomAugspurger?

@TomAugspurger
Copy link
Copy Markdown
Member

That look right @jrbourbeau.

I think we may need to update the tests to something like

if sparse:
    dtype = SparseDtype(dtype, dtype(0))

just before the dtype assertion.

@jrbourbeau
Copy link
Copy Markdown
Member Author

Thanks @TomAugspurger!

@TomAugspurger
Copy link
Copy Markdown
Member

Thanks @jrbourbeau .

@TomAugspurger TomAugspurger merged commit 3c10057 into dask:master Jan 31, 2019
@jrbourbeau jrbourbeau deleted the fix-sklearn-dev branch February 1, 2019 02:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants