From 6eb6c46f04d01bfaced76684323ebbf5236aab9b Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Wed, 6 Nov 2024 14:54:35 -0600
Subject: [PATCH] ignore deprecation warnings from cuda-python, ignore
 cuda-python run_exports

---
 conda/recipes/cuml/meta.yaml | 3 +++
 python/cuml/pyproject.toml   | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml
index 2d7cb645b9..12ed595516 100644
--- a/conda/recipes/cuml/meta.yaml
+++ b/conda/recipes/cuml/meta.yaml
@@ -39,6 +39,7 @@ build:
     - {{ compiler('cuda') }}
     - cuda-cudart-dev
     {% endif %}
+    - cuda-python
 
 requirements:
   build:
@@ -76,8 +77,10 @@ requirements:
     - {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }}
     {% if cuda_major == "11" %}
     - cudatoolkit
+    - cuda-python >=11.7.1,<12.0a0,!=11.8.4
     {% else %}
     - cuda-cudart
+    - cuda-python >=12.0,<13.0a0,!=12.6.1
     {% endif %}
     - cudf ={{ minor_version }}
     - cupy >=12.0.0
diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
index f31f5c61b1..6c46373414 100644
--- a/python/cuml/pyproject.toml
+++ b/python/cuml/pyproject.toml
@@ -45,6 +45,8 @@ filterwarnings = [
   "error:::cudf",
   "ignore:[^.]*ABCs[^.]*:DeprecationWarning:patsy[.*]",
   "ignore:(.*)alias(.*):DeprecationWarning:hdbscan[.*]",
+  # https://github.com/rapidsai/build-planning/issues/116
+  "ignore:.*cuda..* module is deprecated.*:DeprecationWarning",
   # TODO: https://github.com/rapidsai/cuml/issues/5878
   "ignore:.*ndarray.scatter_[(max|add)].* is deprecated:DeprecationWarning:cupyx",
   # TODO: https://github.com/rapidsai/cuml/issues/5879