-
Notifications
You must be signed in to change notification settings - Fork 548
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Ensure all method signatures are sklearn compatible #6260
Ensure all method signatures are sklearn compatible #6260
Conversation
`sklearn` requires `fit`/`fit_transform`/... always take a `y` parameter, even if it's ignored. This adds a test to ensure our signatures match this rule, and fixes any cases where they didn't. This makes it easier to include `cuml` estimators within sklearn pipelines.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Love anything that gets us closer to API compatibility! Nice change and solid implementation.
Failures look unrelated (and maybe transient). I lack permissions to restart the failed runs. |
Merged "main" into this PR to get the changes from #6227 which will solve the failed CI runs |
/merge |
sklearn
requiresfit
/fit_transform
/... always take ay
parameter, even if it's ignored. This adds a test to ensure our signatures match this rule, and fixes any cases where they didn't. This makes it easier to includecuml
estimators within sklearn pipelines.Fixes #6255.