From 7c5c060c456e420c6557a4f554f92b2dfe4f3ccf Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Tue, 25 Aug 2026 15:01:00 -0700 Subject: [PATCH 1/3] Update Treelite to 4.7.1 --- conda/environments/all_cuda-129_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-129_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-133_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-133_arch-x86_64.yaml | 2 +- conda/recipes/cuml/conda_build_config.yaml | 2 +- conda/recipes/libcuml/conda_build_config.yaml | 2 +- cpp/cmake/thirdparty/get_treelite.cmake | 6 +++--- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 4 ++-- 9 files changed, 12 insertions(+), 12 deletions(-) diff --git a/conda/environments/all_cuda-129_arch-aarch64.yaml b/conda/environments/all_cuda-129_arch-aarch64.yaml index 165ea9a27e..b5aa8f66d4 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -84,7 +84,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.7.0,<5.0.0 +- treelite>=4.7.1,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index f0918b00e3..c343de841b 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -83,7 +83,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.7.0,<5.0.0 +- treelite>=4.7.1,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml index da5fe87f62..f47d4a7348 100644 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -84,7 +84,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.7.0,<5.0.0 +- treelite>=4.7.1,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml index 28e7dba306..ca3b7d2086 100644 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -83,7 +83,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.7.0,<5.0.0 +- treelite>=4.7.1,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/recipes/cuml/conda_build_config.yaml b/conda/recipes/cuml/conda_build_config.yaml index 0b101ca8fd..2a0c083cfd 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.7.0,<5.0.0" + - ">=4.7.1,<5.0.0" diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml index 5780e54c99..fc7436dcad 100644 --- a/conda/recipes/libcuml/conda_build_config.yaml +++ b/conda/recipes/libcuml/conda_build_config.yaml @@ -17,4 +17,4 @@ cmake_version: - ">=4.0" treelite_version: - - ">=4.7.0,<5.0.0" + - ">=4.7.1,<5.0.0" diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake index 37727e2d4e..5e337c8735 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-2026, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # 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.7.0 - PINNED_TAG 74b25ecedb964ccac37d034860cc5c1224e73e91 +find_and_configure_treelite(VERSION 4.7.1 + PINNED_TAG 3a44165f6c172d24a884591c544d9557b35c778e EXCLUDE_FROM_ALL ${CUML_EXCLUDE_TREELITE_FROM_ALL} BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC}) diff --git a/dependencies.yaml b/dependencies.yaml index 30911ade9c..aaae184c9e 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -351,7 +351,7 @@ dependencies: packages: # skipping Cython 3.2.7 because of https://github.com/cython/cython/issues/7781 - &cython cython>=3.2.2,!=3.2.7 - - &treelite treelite>=4.7.0,<5.0.0 + - &treelite treelite>=4.7.1,<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 7d68bbee50..5ec4b201fb 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -102,7 +102,7 @@ dependencies = [ "rmm==26.10.*,>=0.0.0a0", "scikit-learn>=1.6", "scipy>=1.14.0", - "treelite>=4.7.0,<5.0.0", + "treelite>=4.7.1,<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", @@ -202,7 +202,7 @@ requires = [ "nvforest==26.10.*,>=0.0.0a0", "pylibraft==26.10.*,>=0.0.0a0", "rmm==26.10.*,>=0.0.0a0", - "treelite>=4.7.0,<5.0.0", + "treelite>=4.7.1,<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 242eb77ad35f2ac377581b3725a1604cea9e2e63 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Tue, 1 Sep 2026 18:52:48 -0700 Subject: [PATCH 2/3] Use Treelite 4.7.2 --- conda/environments/all_cuda-129_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-129_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-133_arch-aarch64.yaml | 2 +- conda/environments/all_cuda-133_arch-x86_64.yaml | 2 +- cpp/cmake/thirdparty/get_treelite.cmake | 4 ++-- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 4 ++-- 7 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 b5aa8f66d4..ffcccab8d2 100644 --- a/conda/environments/all_cuda-129_arch-aarch64.yaml +++ b/conda/environments/all_cuda-129_arch-aarch64.yaml @@ -84,7 +84,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.7.1,<5.0.0 +- treelite>=4.7.2,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-129_arch-x86_64.yaml b/conda/environments/all_cuda-129_arch-x86_64.yaml index c343de841b..4a781091df 100644 --- a/conda/environments/all_cuda-129_arch-x86_64.yaml +++ b/conda/environments/all_cuda-129_arch-x86_64.yaml @@ -83,7 +83,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.7.1,<5.0.0 +- treelite>=4.7.2,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-133_arch-aarch64.yaml b/conda/environments/all_cuda-133_arch-aarch64.yaml index f47d4a7348..cabab5ef63 100644 --- a/conda/environments/all_cuda-133_arch-aarch64.yaml +++ b/conda/environments/all_cuda-133_arch-aarch64.yaml @@ -84,7 +84,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-aarch64==2.28 -- treelite>=4.7.1,<5.0.0 +- treelite>=4.7.2,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/conda/environments/all_cuda-133_arch-x86_64.yaml b/conda/environments/all_cuda-133_arch-x86_64.yaml index ca3b7d2086..1be35700f1 100644 --- a/conda/environments/all_cuda-133_arch-x86_64.yaml +++ b/conda/environments/all_cuda-133_arch-x86_64.yaml @@ -83,7 +83,7 @@ dependencies: - sphinx-markdown-tables - statsmodels - sysroot_linux-64==2.28 -- treelite>=4.7.1,<5.0.0 +- treelite>=4.7.2,<5.0.0 - umap-learn>=0.5.7,<0.5.13 - xgboost>=3.3.0 - pip: diff --git a/cpp/cmake/thirdparty/get_treelite.cmake b/cpp/cmake/thirdparty/get_treelite.cmake index 5e337c8735..a831e9fb66 100644 --- a/cpp/cmake/thirdparty/get_treelite.cmake +++ b/cpp/cmake/thirdparty/get_treelite.cmake @@ -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.7.1 - PINNED_TAG 3a44165f6c172d24a884591c544d9557b35c778e +find_and_configure_treelite(VERSION 4.7.2 + PINNED_TAG e8a2a7770354318f1e702ba8c059c3b6096d78cf EXCLUDE_FROM_ALL ${CUML_EXCLUDE_TREELITE_FROM_ALL} BUILD_STATIC_LIBS ${CUML_USE_TREELITE_STATIC}) diff --git a/dependencies.yaml b/dependencies.yaml index aaae184c9e..bfc721da87 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -351,7 +351,7 @@ dependencies: packages: # skipping Cython 3.2.7 because of https://github.com/cython/cython/issues/7781 - &cython cython>=3.2.2,!=3.2.7 - - &treelite treelite>=4.7.1,<5.0.0 + - &treelite treelite>=4.7.2,<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 5ec4b201fb..ac9a10bb14 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -102,7 +102,7 @@ dependencies = [ "rmm==26.10.*,>=0.0.0a0", "scikit-learn>=1.6", "scipy>=1.14.0", - "treelite>=4.7.1,<5.0.0", + "treelite>=4.7.2,<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", @@ -202,7 +202,7 @@ requires = [ "nvforest==26.10.*,>=0.0.0a0", "pylibraft==26.10.*,>=0.0.0a0", "rmm==26.10.*,>=0.0.0a0", - "treelite>=4.7.1,<5.0.0", + "treelite>=4.7.2,<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 5e5fa54c29ebda173fcd289aacb096053fdad0c3 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Tue, 1 Sep 2026 18:58:32 -0700 Subject: [PATCH 3/3] Update conda_build_config.yaml --- conda/recipes/cuml/conda_build_config.yaml | 2 +- conda/recipes/libcuml/conda_build_config.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/conda/recipes/cuml/conda_build_config.yaml b/conda/recipes/cuml/conda_build_config.yaml index 2a0c083cfd..f7454ea5c6 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.7.1,<5.0.0" + - ">=4.7.2,<5.0.0" diff --git a/conda/recipes/libcuml/conda_build_config.yaml b/conda/recipes/libcuml/conda_build_config.yaml index fc7436dcad..ff84b755a0 100644 --- a/conda/recipes/libcuml/conda_build_config.yaml +++ b/conda/recipes/libcuml/conda_build_config.yaml @@ -17,4 +17,4 @@ cmake_version: - ">=4.0" treelite_version: - - ">=4.7.1,<5.0.0" + - ">=4.7.2,<5.0.0"