Exclude XGBoost on 3.11 test runs

dependencies.yaml

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 # Dependency list for https://github.com/rapidsai/dependency-file-generator
@@ -30,6 +30,9 @@ files:
     output: none
     includes:
       # "devcontainers" includes everything in "all", excluding test_python_xgboost
+      # The libxgboost package depends on librmm but we do not want to have a
+      # package depending on librmm in devcontainers since it should be built
+      # from source.
       - common_build
       - cuda
       - cuda_version
@@ -423,17 +426,35 @@ dependencies:
           - pytest-xdist
           - *scikit_learn
   test_python_xgboost:
-    common:
+    specific:
       - output_types: [conda]
-        packages:
-          # We must separate xgboost into its own list so that it is not
-          # included in the "devcontainers" key. The libxgboost package depends
-          # on librmm but we do not want to have a package depending on librmm
-          # in devcontainers since it should be built from source.
-          - rapids-xgboost==26.8.*,>=0.0.0a0
+        matrices:
+        - matrix:
+            py: "3.11"
+          packages: []
+        - matrix:
+          packages:
+            - rapids-xgboost==26.8.*,>=0.0.0a0
       - output_types: [requirements, pyproject]
-        packages:
-          - xgboost>=2.1.0
+        matrices:
+          - matrix:
+              py: "3.11"
+            packages: []
+          - matrix:
+              cuda: "12.*"
+              cuda_suffixed: "true"
+            packages:
+              - xgboost-cu12>=3.3.0; python_version >= '3.12'
+          - matrix:
+              cuda: "13.*"
+              cuda_suffixed: "true"
+            packages:
+              - xgboost-cu13>=3.3.0; python_version >= '3.12'
+          # this unsuffixed dependency is useful in environments where xgboost is built from source
+          # (like RAPIDS DLFW builds)
+          - matrix:
+            packages:
+              - xgboost>=3.3.0; python_version >= '3.12'
   depends_on_cuda_python:
     specific:
       - output_types: [conda, requirements, pyproject]

python/nvforest/pyproject.toml

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 [build-system]
@@ -51,7 +51,7 @@ test = [
     "pytest-cov",
     "pytest-xdist",
     "scikit-learn>=1.5",
-    "xgboost>=2.1.0",
+    "xgboost>=3.3.0; python_version >= '3.12'",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]

python/nvforest/tests/test_benchmark.py

@@ -1,5 +1,5 @@
 #
-# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 
@@ -323,7 +323,7 @@ def test_cli_dry_run_quick_test(self):
         # Quick test has fewer configurations
         assert any(
             f"Total benchmark configurations: {x}" in result.output
-            for x in [8, 12]
+            for x in [4, 8, 12]
         )
 
     @pytest.mark.unit