Skip to content

Commit

Permalink
Bump Treelite to 4.2.1 (#5908)
Browse files Browse the repository at this point in the history
Treelite 4.2.1 contains the following improvements:

* Compatibility patch for latest RapidJSON (dmlc/treelite#567)
* Support for NumPy 2.0 (dmlc/treelite#562). Thanks @jameslamb
* Handle certain class of XGBoost models (dmlc/treelite#564)

Authors:
  - Philip Hyunsu Cho (https://github.com/hcho3)

Approvers:
  - Bradley Dice (https://github.com/bdice)
  - Dante Gama Dessavre (https://github.com/dantegd)
  - Ray Douglass (https://github.com/raydouglass)
  - James Lamb (https://github.com/jameslamb)

URL: #5908
  • Loading branch information
hcho3 authored Jun 15, 2024
1 parent 07a3fb9 commit 205748c
Show file tree
Hide file tree
Showing 7 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion conda/environments/all_cuda-118_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ dependencies:
- sphinx<6
- statsmodels
- sysroot_linux-64==2.17
- treelite==4.1.2
- treelite==4.2.1
- umap-learn==0.5.3
- pip:
- dask-glm==0.3.0
Expand Down
2 changes: 1 addition & 1 deletion conda/environments/all_cuda-122_arch-x86_64.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ dependencies:
- sphinx<6
- statsmodels
- sysroot_linux-64==2.17
- treelite==4.1.2
- treelite==4.2.1
- umap-learn==0.5.3
- pip:
- dask-glm==0.3.0
Expand Down
2 changes: 1 addition & 1 deletion conda/recipes/cuml/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,4 @@ c_stdlib_version:
- "=2.17"

treelite_version:
- "=4.1.2"
- "=4.2.1"
2 changes: 1 addition & 1 deletion conda/recipes/libcuml/conda_build_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ cmake_version:
- ">=3.26.4"

treelite_version:
- "=4.1.2"
- "=4.2.1"

# The CTK libraries below are missing from the conda-forge::cudatoolkit package
# for CUDA 11. The "*_host_*" version specifiers correspond to `11.8` packages
Expand Down
4 changes: 2 additions & 2 deletions cpp/cmake/thirdparty/get_treelite.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ function(find_and_configure_treelite)
rapids_export_find_package_root(BUILD Treelite [=[${CMAKE_CURRENT_LIST_DIR}]=] EXPORT_SET cuml-exports)
endfunction()

find_and_configure_treelite(VERSION 4.1.2
PINNED_TAG 454abc4bfe03e6e9d45155c15f74ae36d8c73111
find_and_configure_treelite(VERSION 4.2.1
PINNED_TAG 07405e7b8d79e84f7e7ab29bb8b00de20dc6aed4
EXCLUDE_FROM_ALL ${CUML_EXCLUDE_TREELITE_FROM_ALL}
BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC})
2 changes: 1 addition & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,7 @@ dependencies:
- output_types: [conda, requirements, pyproject]
packages:
- cython>=3.0.0
- &treelite treelite==4.1.2
- &treelite treelite==4.2.1
- output_types: conda
packages:
- &pylibraft_conda pylibraft==24.8.*,>=0.0.0a0
Expand Down
4 changes: 2 additions & 2 deletions python/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ dependencies = [
"rapids-dask-dependency==24.8.*,>=0.0.0a0",
"rmm==24.8.*,>=0.0.0a0",
"scipy>=1.8.0",
"treelite==4.1.2",
"treelite==4.2.1",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../dependencies.yaml and run `rapids-dependency-file-generator`.
classifiers = [
"Intended Audience :: Developers",
Expand Down Expand Up @@ -159,7 +159,7 @@ requires = [
"ninja",
"pylibraft==24.8.*,>=0.0.0a0",
"rmm==24.8.*,>=0.0.0a0",
"treelite==4.1.2",
"treelite==4.2.1",
] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../dependencies.yaml and run `rapids-dependency-file-generator`.
dependencies-file = "../dependencies.yaml"

Expand Down

0 comments on commit 205748c

Please sign in to comment.