Skip to content

Conversation

@RAMitchell
Copy link
Contributor

sklearn 1.6 is now the default on conda forge so I have updated legate-boost to use >=1.6

I've not taken the xgboost approach of continuing to support sklearn 1.5 as legate-boost does not currently have to worry about breaking changes for existing users and it would create additional work to test both in CI.
dmlc/xgboost#11021

@RAMitchell RAMitchell requested a review from jameslamb December 13, 2024 11:14
Copy link
Member

@jameslamb jameslamb left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it's totally fair at this stage in this project to only support 1 minor version of scikit-learn, agree that it doesn't have to have the same wide compatibility as XGBoost.

And this set of changes makes sense to me based on what I've experienced adding scikit-learn 1.6 support to other libraries.

Left some very very minor comments, do whatever you want with them, and I don't need to re-review.

return X


def lb_check_X_y(X: Any, y: Any = None) -> Any:
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
def lb_check_X_y(X: Any, y: Any = None) -> Any:
def _lb_check_X_y(X: Any, y: Any = None) -> Any:

I have a slight preference for prefixing this with _ to further discourage anyone from importing and relying on it, since it's really an internal implementation detail of how this project stays compatible with scikit-learn. (ref: #143)

@RAMitchell RAMitchell merged commit 98412e3 into rapidsai:main Dec 16, 2024
10 checks passed
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