Use RAPIDS XGBoost 2.1.4

rapidsai · Feb 7, 2025 · 7e1f360 · 7e1f360
1 parent 847a76b
commit 7e1f360
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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]