Skip to content

Remove dask to sparse workarounds#6489

Merged
rapids-bot[bot] merged 2 commits intorapidsai:branch-25.04from
TomAugspurger:tom/fix/to_sparse_dask_array
Mar 25, 2025
Merged

Remove dask to sparse workarounds#6489
rapids-bot[bot] merged 2 commits intorapidsai:branch-25.04from
TomAugspurger:tom/fix/to_sparse_dask_array

Conversation

@TomAugspurger
Copy link
Copy Markdown
Contributor

This fixes a failure in test_to_sparse_dask_array with dask main. It seems that the workarounds previously implemented are fixed in cupy / dask and can now be removed from cuml.

xref rapidsai/dask-upstream-testing#37, specifically the failure here.

Not sure if anyone has the context to say for sure, but I'm curious how well we think the existing test suite would catch any regressions here. I haven't done any kind of performance / memory profiling to make sure there aren't any more subtle regressions.

@TomAugspurger TomAugspurger requested a review from a team as a code owner March 25, 2025 16:48
@TomAugspurger TomAugspurger requested review from betatim and vyasr March 25, 2025 16:48
@github-actions github-actions Bot added the Cython / Python Cython or Python issue label Mar 25, 2025
This fixes a failure in `test_to_sparse_dask_array` with dask main.
It seems that the workarounds previously implemented are fixed in cupy /
dask and can now be removed from cuml.

xref rapidsai/dask-upstream-testing#37
@jcrist jcrist added bug Something isn't working non-breaking Non-breaking change labels Mar 25, 2025
@jcrist
Copy link
Copy Markdown
Member

jcrist commented Mar 25, 2025

Thanks Tom! Looks like you have some flake8 errors, but otherwise LGTM.

@TomAugspurger
Copy link
Copy Markdown
Contributor Author

Thanks, the flake8 errors should be fixed now.

https://github.com/rapidsai/cuml/actions/runs/14066201872/job/39390623623?pr=6489#step:9:1486 has a failure in test_rf_classification_seed that I'll assume is unrelated to this PR.

@jcrist
Copy link
Copy Markdown
Member

jcrist commented Mar 25, 2025

Yep, that's a known flaky test failure. See #6372.

@jcrist
Copy link
Copy Markdown
Member

jcrist commented Mar 25, 2025

/merge

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working Cython / Python Cython or Python issue non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants