From d0476891ed2ef500543c59d3c26c96e41e646739 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Fri, 6 Mar 2026 20:57:51 -0800 Subject: [PATCH 1/2] Update Treelite to 4.7.0 --- conda/environments/all_cuda-129_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-129_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-131_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-131_arch-x86_64.yaml | 2 +- conda/recipes/cuml/conda_build_config.yaml | 2 +- conda/recipes/libcuml/conda_build_config.yaml | 2 +- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 4 ++-- 8 files changed, 9 insertions(+), 9 deletions(-) diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index a9dd782200..410598097f 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -80,6 +80,6 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.6.1,<5.0.0 +- treelite>=4.7.0,<5.0.0 - umap-learn>=0.5.7,<0.5.12 name: all_cuda-129_arch-aarch64 diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index e1fb3593aa..87164b433d 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -80,6 +80,6 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.6.1,<5.0.0 +- treelite>=4.7.0,<5.0.0 - umap-learn>=0.5.7,<0.5.12 name: all_cuda-129_arch-x86_64 diff --git a/conda/environments/all_cuda-131_arch-aarch64.yaml b/conda/environments/all_cuda-131_arch-aarch64.yaml index 74d531c5d1..22f2b591aa 100644 --- a/conda/environments/all_cuda-131_arch-aarch64.yaml +++ b/conda/environments/all_cuda-131_arch-aarch64.yaml @@ -80,6 +80,6 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.6.1,<5.0.0 +- treelite>=4.7.0,<5.0.0 - umap-learn>=0.5.7,<0.5.12 name: all_cuda-131_arch-aarch64 diff --git a/conda/environments/all_cuda-131_arch-x86_64.yaml b/conda/environments/all_cuda-131_arch-x86_64.yaml index 2e788184a9..3603369d4f 100644 --- a/conda/environments/all_cuda-131_arch-x86_64.yaml +++ b/conda/environments/all_cuda-131_arch-x86_64.yaml @@ -80,6 +80,6 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.6.1,<5.0.0 +- treelite>=4.7.0,<5.0.0 - umap-learn>=0.5.7,<0.5.12 name: all_cuda-131_arch-x86_64 diff --git a/conda/recipes/cuml/conda_build_config.yaml b/conda/recipes/cuml/conda_build_config.yaml index 6e08076a8f..6ccc1e5035 100644 --- a/conda/recipes/cuml/conda_build_config.yaml +++ b/conda/recipes/cuml/conda_build_config.yaml @@ -17,4 +17,4 @@ c_stdlib_version: - "=2.28" treelite_version: - - ">=4.6.1,<5.0.0" + - ">=4.7.0,<5.0.0" diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml index c62869acd6..6ba84c9d58 100644 --- a/conda/recipes/libcuml/conda_build_config.yaml +++ b/conda/recipes/libcuml/conda_build_config.yaml @@ -17,4 +17,4 @@ cmake_version: - ">=3.30.4" treelite_version: - - ">=4.6.1,<5.0.0" + - ">=4.7.0,<5.0.0" diff --git a/dependencies.yaml b/dependencies.yaml index c3deee9d3a..81509e5a98 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -316,7 +316,7 @@ dependencies: - output_types: [conda, requirements, pyproject] packages: - &cython cython>=3.2.2 - - &treelite treelite>=4.6.1,<5.0.0 + - &treelite treelite>=4.7.0,<5.0.0 py_run_cuml: common: - output_types: [conda, requirements, pyproject] diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml index aa5e4ede11..809b58bcf9 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -95,7 +95,7 @@ dependencies = [ "rmm==26.4.*,>=0.0.0a0", "scikit-learn>=1.5", "scipy>=1.13.0", - "treelite>=4.6.1,<5.0.0", + "treelite>=4.7.0,<5.0.0", ] # 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", @@ -191,7 +191,7 @@ requires = [ "ninja", "pylibraft==26.4.*,>=0.0.0a0", "rmm==26.4.*,>=0.0.0a0", - "treelite>=4.6.1,<5.0.0", + "treelite>=4.7.0,<5.0.0", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. [tool.scikit-build] From 3d9fd513f8f7ec96c664061388faf56cfaf8cbf3 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Sat, 7 Mar 2026 15:13:08 -0800 Subject: [PATCH 2/2] Update get_treelite.cmake --- cpp/cmake/thirdparty/get_treelite.cmake | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake index 1c5c1809a6..37727e2d4e 100644 --- a/cpp/cmake/thirdparty/get_treelite.cmake +++ b/cpp/cmake/thirdparty/get_treelite.cmake @@ -1,6 +1,6 @@ #============================================================================= # cmake-format: off -# SPDX-FileCopyrightText: Copyright (c) 2021-2025, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION. # SPDX-License-Identifier: Apache-2.0 # cmake-format: on #============================================================================= @@ -70,7 +70,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.6.1 - PINNED_TAG e14222bfdf02cba3ece0f3692f3574711cab0fc2 +find_and_configure_treelite(VERSION 4.7.0 + PINNED_TAG 74b25ecedb964ccac37d034860cc5c1224e73e91 EXCLUDE_FROM_ALL ${CUML_EXCLUDE_TREELITE_FROM_ALL} BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC})