-
Notifications
You must be signed in to change notification settings - Fork 538
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
Update FIL tests to use XGBoost UBJSON instead of binary #6153
Conversation
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.
Any reason why we cant use xgboost>=2.1.3
?
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.
Thanks Hyunsu! 🙏
Had a recommendation below
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.
Looks good. Thanks Hyunsu! 🙏
Checked in offline. Sounds like we are ready to merge this. Going to queue it to merge so we don't forget (as code freeze is around the corner) |
/merge |
Starting from 2.1, XGBoost uses UBJSON format to serialize models.
Replace all uses of the legacy model format with UBJSON.
Also make
xgboost
a test dependency of cuML so that the FIL tests run in the CI pipelines.