Unblock CI for 25.04

conda/environments/all_cuda-118_arch-aarch64.yaml

@@ -77,5 +77,5 @@ dependencies:
 - sysroot_linux-aarch64==2.28
 - treelite==4.4.1
 - umap-learn==0.5.6
-- xgboost>=2.1.0
+- xgboost>=2.1.0,<3.0.0
 name: all_cuda-118_arch-aarch64

conda/environments/all_cuda-118_arch-x86_64.yaml

@@ -77,5 +77,5 @@ dependencies:
 - sysroot_linux-64==2.28
 - treelite==4.4.1
 - umap-learn==0.5.6
-- xgboost>=2.1.0
+- xgboost>=2.1.0,<3.0.0
 name: all_cuda-118_arch-x86_64

conda/environments/all_cuda-128_arch-aarch64.yaml

@@ -73,5 +73,5 @@ dependencies:
 - sysroot_linux-aarch64==2.28
 - treelite==4.4.1
 - umap-learn==0.5.6
-- xgboost>=2.1.0
+- xgboost>=2.1.0,<3.0.0
 name: all_cuda-128_arch-aarch64

conda/environments/all_cuda-128_arch-x86_64.yaml

@@ -73,5 +73,5 @@ dependencies:
 - sysroot_linux-64==2.28
 - treelite==4.4.1
 - umap-learn==0.5.6
-- xgboost>=2.1.0
+- xgboost>=2.1.0,<3.0.0
 name: all_cuda-128_arch-x86_64

dependencies.yaml

@@ -493,7 +493,7 @@ dependencies:
           - pytest-xdist
           - seaborn
           - *scikit_learn
-          - &xgboost xgboost>=2.1.0
+          - &xgboost xgboost>=2.1.0,<3.0.0
           - statsmodels
           - umap-learn==0.5.6
           - pynndescent

python/cuml/cuml/tests/test_device_selection.py

@@ -72,6 +72,7 @@
 from cuml.internals.safe_imports import cpu_only_import
 
 np = cpu_only_import("numpy")
+cp = gpu_only_import("cupy")
 pd = cpu_only_import("pandas")
 cudf = gpu_only_import("cudf")
 
@@ -1121,6 +1122,14 @@ def test_svc_methods(
     class_type,
     probability,
 ):
+    # Skip on CUDA 11.8 due to segfaults in decision_function
+    # See: https://github.com/rapidsai/cuml/issues/6480
+    if 11080 <= cp.cuda.runtime.runtimeGetVersion() < 11090:
+        pytest.skip(
+            "Skipping test_svc_methods on CUDA 11.8 due to segfaults in "
+            "decision_function (#6480)"
+        )
+
     if class_type == "single_class":
         X_train = X_train_class
         y_train = y_train_class

python/cuml/pyproject.toml

@@ -142,7 +142,7 @@ test = [
     "seaborn",
     "statsmodels",
     "umap-learn==0.5.6",
-    "xgboost>=2.1.0",
+    "xgboost>=2.1.0,<3.0.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]