Pinned joblib library to 1.1.0 in the XGBoost components. Previous versions of joblib had an issue that caused the XGBoost model to be loaded incorrectly
Fixes #26
How has this been tested?
Tested by running XGBoost training and prediction pipelines on Vertex
E2E tests now passing on equivalent PR here
Checklist
- [ ] I have commented my code, particularly in hard-to-understand areas
[x] I have successfully run the E2E tests
- [ ] I have added tests that prove my fix is effective or that my feature works
[x] New and existing unit tests pass locally with my changes
- [ ] I have updated any relevant documentation to reflect my changes
ghost
commented
1 year ago
Description
Pinned
jobliblibrary to1.1.0in the XGBoost components. Previous versions ofjoblibhad an issue that caused the XGBoost model to be loaded incorrectlyFixes #26
How has this been tested?
Tested by running XGBoost training and prediction pipelines on Vertex E2E tests now passing on equivalent PR here
Checklist
- [ ] I have commented my code, particularly in hard-to-understand areas- [ ] I have added tests that prove my fix is effective or that my feature works- [ ] I have updated any relevant documentation to reflect my changes