From fce4fcf20de3d28808a3dcb00d092070f44d8292 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Fri, 6 Dec 2024 13:20:09 -0800
Subject: [PATCH] move CUDA libs deps to wheels, ignore deprecation warning

---
 dependencies.yaml                  | 2 +-
 python/cugraph/pytest.ini          | 1 +
 python/libcugraph/pyproject.toml   | 4 ++++
 python/pylibcugraph/pyproject.toml | 4 ----
 4 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index a3ab343f07..51a0f5e0c2 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -134,6 +134,7 @@ files:
     extras:
       table: project
     includes:
+      - cuda_wheels
       - depends_on_rmm
   py_build_pylibcugraph:
     output: pyproject
@@ -161,7 +162,6 @@ files:
     extras:
       table: project
     includes:
-      - cuda_wheels
       - depends_on_libcugraph
       - depends_on_rmm
       - depends_on_pylibraft
diff --git a/python/cugraph/pytest.ini b/python/cugraph/pytest.ini
index bf6e6bdd80..7bdef2d977 100644
--- a/python/cugraph/pytest.ini
+++ b/python/cugraph/pytest.ini
@@ -62,6 +62,7 @@ filterwarnings =
           error::FutureWarning
           error::DeprecationWarning
           # TODO
+          ignore:.*cuda..* module is deprecated.*:DeprecationWarning
           ignore:Multi is deprecated and the removal of multi edges will no longer be supported:FutureWarning
           ignore:The legacy column names:FutureWarning
           ignore:The include_hop_column flag is deprecated and will be removed:FutureWarning
diff --git a/python/libcugraph/pyproject.toml b/python/libcugraph/pyproject.toml
index 5f680c44ee..4f4b9ebbb3 100644
--- a/python/libcugraph/pyproject.toml
+++ b/python/libcugraph/pyproject.toml
@@ -39,6 +39,10 @@ classifiers = [
     "Environment :: GPU :: NVIDIA CUDA",
 ]
 dependencies = [
+    "nvidia-cublas",
+    "nvidia-curand",
+    "nvidia-cusolver",
+    "nvidia-cusparse",
     "rmm==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
diff --git a/python/pylibcugraph/pyproject.toml b/python/pylibcugraph/pyproject.toml
index 5f7c7dd409..b0175b16ed 100644
--- a/python/pylibcugraph/pyproject.toml
+++ b/python/pylibcugraph/pyproject.toml
@@ -24,10 +24,6 @@ license = { text = "Apache 2.0" }
 requires-python = ">=3.10"
 dependencies = [
     "libcugraph==25.2.*,>=0.0.0a0",
-    "nvidia-cublas",
-    "nvidia-curand",
-    "nvidia-cusolver",
-    "nvidia-cusparse",
     "pylibraft==25.2.*,>=0.0.0a0",
     "rmm==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.