From 7e1f360540633ad48718a99f0071ae5877bfe4d4 Mon Sep 17 00:00:00 2001 From: jakirkham Date: Thu, 6 Feb 2025 23:07:19 -0800 Subject: [PATCH] Use RAPIDS XGBoost 2.1.4 --- conda/environments/all_cuda-118_arch-x86_64.yaml | 2 +- conda/environments/all_cuda-128_arch-x86_64.yaml | 2 +- dependencies.yaml | 2 +- python/cuml/pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 45b90794a5..bf538e9b59 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -78,5 +78,5 @@ dependencies: - sysroot_linux-64==2.28 - treelite==4.4.1 - umap-learn==0.5.6 -- xgboost>=2.1.0 +- xgboost==2.1.4 name: all_cuda-118_arch-x86_64 diff --git a/conda/environments/all_cuda-128_arch-x86_64.yaml b/conda/environments/all_cuda-128_arch-x86_64.yaml index fc08dacfd7..7b0bee9dc4 100644 --- a/conda/environments/all_cuda-128_arch-x86_64.yaml +++ b/conda/environments/all_cuda-128_arch-x86_64.yaml @@ -74,5 +74,5 @@ dependencies: - sysroot_linux-64==2.28 - treelite==4.4.1 - umap-learn==0.5.6 -- xgboost>=2.1.0 +- xgboost==2.1.4 name: all_cuda-128_arch-x86_64 diff --git a/dependencies.yaml b/dependencies.yaml index 968ebc5f31..93151cd084 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -489,7 +489,7 @@ dependencies: - pytest-xdist - seaborn - *scikit_learn - - &xgboost xgboost>=2.1.0 + - &xgboost xgboost==2.1.4 - statsmodels - umap-learn==0.5.6 - pynndescent diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml index 7d9bf1d541..932900b7a6 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -143,7 +143,7 @@ test = [ "seaborn", "statsmodels", "umap-learn==0.5.6", - "xgboost>=2.1.0", + "xgboost==2.1.4", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. [project.urls]