search

dmlc / treelite

Universal model exchange and serialization format for decision tree forests
https://treelite.readthedocs.io/en/latest/
Apache License 2.0
723 stars 98 forks source link

scikit-learn 1.4.0 breaks unit test #544

Closed hcho3 closed 4 months ago

hcho3 commented 5 months ago

Affected tests:

tests/python/test_sklearn_integration.py::test_skl_hist_gradient_boosting_with_categorical
tests/serializer/test_serializer.py::test_serialize_as_buffer[GradientBoostingClassifier]

(https://github.com/dmlc/treelite/actions/runs/7737603234)

The tests pass with scikit-learn 1.3.2.

hcho3 commented 5 months ago

Update: tests/serializer/test_serializer.py::test_serialize_as_buffer[GradientBoostingClassifier] is fixed by #542