WIP: pylibcugraph: declare cupy and numpy hard dependencies

conda/recipes/pylibcugraph/meta.yaml

@@ -1,4 +1,4 @@
-# Copyright (c) 2023-2024, NVIDIA CORPORATION.
+# Copyright (c) 2023-2025, NVIDIA CORPORATION.
 
 {% set version = environ['RAPIDS_PACKAGE_VERSION'].lstrip('v') + environ.get('VERSION_SUFFIX', '') %}
 {% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
@@ -74,7 +74,9 @@ requirements:
     {% else %}
     - cuda-cudart
     {% endif %}
+    - cupy >=12.0.0
     - libcugraph ={{ version }}
+    - numpy>=1.23,<3.0a0
     - pylibraft ={{ minor_version }}
     - python
     - rmm ={{ minor_version }}

dependencies.yaml

@@ -18,6 +18,7 @@ files:
       - depends_on_libcudf
       - depends_on_libraft
       - depends_on_librmm
+      - depends_on_numpy
       - depends_on_pylibraft
       - depends_on_pylibwholegraph
       - depends_on_pytorch
@@ -33,7 +34,6 @@ files:
       - test_notebook
       - test_python_common
       - test_python_cugraph
-      - test_python_pylibcugraph
   checks:
     output: none
     includes:
@@ -64,12 +64,12 @@ files:
     includes:
       - cuda_version
       - depends_on_cudf
+      - depends_on_numpy
       - depends_on_pylibwholegraph
       - depends_on_pytorch
       - py_version
       - test_python_common
       - test_python_cugraph
-      - test_python_pylibcugraph
   py_build_cugraph:
     output: pyproject
     pyproject_dir: python/cugraph
@@ -141,6 +141,8 @@ files:
       table: project
     includes:
       - cuda_wheels
+      - depends_on_cupy
+      - depends_on_numpy
       - depends_on_pylibraft
       - depends_on_rmm
   py_test_pylibcugraph:
@@ -151,8 +153,8 @@ files:
       key: test
     includes:
       - depends_on_cudf
+      - depends_on_numpy
       - test_python_common
-      - test_python_pylibcugraph
   py_build_cugraph_service_client:
     output: pyproject
     pyproject_dir: python/cugraph-service/client
@@ -188,6 +190,7 @@ files:
       - depends_on_cupy
       - depends_on_dask_cuda
       - depends_on_dask_cudf
+      - depends_on_numpy
       - depends_on_rmm
       - depends_on_ucx_py
       - python_run_cugraph_service_server
@@ -398,7 +401,6 @@ dependencies:
         packages:
           - &dask rapids-dask-dependency==25.2.*,>=0.0.0a0
           - &numba numba>=0.57
-          - &numpy numpy>=1.23,<3.0a0
       - output_types: conda
         packages:
           - aiohttp
@@ -432,7 +434,6 @@ dependencies:
         packages:
           - *dask
           - *numba
-          - *numpy
           - *thrift
   test_cpp:
     common:
@@ -462,17 +463,11 @@ dependencies:
         packages:
           - certifi
           - networkx>=2.5.1
-          - *numpy
           - python-louvain
           - scikit-learn>=0.23.1
       - output_types: [conda]
         packages:
           - *thrift
-  test_python_pylibcugraph:
-    common:
-      - output_types: [conda, pyproject]
-        packages:
-          - *numpy
 
   depends_on_cugraph:
     common:
@@ -765,6 +760,12 @@ dependencies:
               - cupy-cuda11x>=12.0.0
           - {matrix: null, packages: *cupy_packages_cu11}
 
+  depends_on_numpy:
+    common:
+      - output_types: [conda, pyproject, requirements]
+        packages:
+          - numpy>=1.23,<3.0a0
+
   depends_on_ucx_py:
     common:
       - output_types: conda

python/cugraph-service/server/pyproject.toml

@@ -48,7 +48,6 @@ cugraph-service-server = "cugraph_service_server.__main__:main"
 test = [
     "certifi",
     "networkx>=2.5.1",
-    "numpy>=1.23,<3.0a0",
     "pandas",
     "pytest",
     "pytest-benchmark",

python/cugraph/pyproject.toml

@@ -30,7 +30,6 @@ dependencies = [
     "dask-cudf==25.2.*,>=0.0.0a0",
     "fsspec[http]>=0.6.0",
     "numba>=0.57",
-    "numpy>=1.23,<3.0a0",
     "pylibcugraph==25.2.*,>=0.0.0a0",
     "raft-dask==25.2.*,>=0.0.0a0",
     "rapids-dask-dependency==25.2.*,>=0.0.0a0",
@@ -49,7 +48,6 @@ classifiers = [
 test = [
     "certifi",
     "networkx>=2.5.1",
-    "numpy>=1.23,<3.0a0",
     "pandas",
     "pylibwholegraph==25.2.*,>=0.0.0a0",
     "pytest",

python/pylibcugraph/pylibcugraph/bfs.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -14,6 +14,8 @@
 # Have cython use python 3 syntax
 # cython: language_level = 3
 
+import cupy
+
 from libc.stdint cimport uintptr_t
 from libc.stdint cimport int32_t
 from libc.limits cimport INT_MAX
@@ -141,11 +143,6 @@ def bfs(ResourceHandle handle, _GPUGraph graph,
     >>> })
     """
 
-    try:
-        import cupy
-    except ModuleNotFoundError:
-        raise RuntimeError("bfs requires the cupy package, which could not "
-                           "be imported")
     assert_CAI_type(sources, "sources")
 
     if depth_limit <= 0:

python/pylibcugraph/pylibcugraph/node2vec.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -14,6 +14,8 @@
 # Have cython use python 3 syntax
 # cython: language_level = 3
 
+import cupy
+
 from libc.stdint cimport uintptr_t
 
 from pylibcugraph._cugraph_c.types cimport (
@@ -124,14 +126,6 @@ def node2vec(ResourceHandle resource_handle,
 
     """
 
-    # FIXME: import these modules here for now until a better pattern can be
-    # used for optional imports (perhaps 'import_optional()' from cugraph), or
-    # these are made hard dependencies.
-    try:
-        import cupy
-    except ModuleNotFoundError:
-        raise RuntimeError("node2vec requires the cupy package, which could not "
-                           "be imported")
     assert_CAI_type(seed_array, "seed_array")
 
     cdef cugraph_resource_handle_t* c_resource_handle_ptr = \

python/pylibcugraph/pylibcugraph/pagerank.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -14,6 +14,9 @@
 # Have cython use python 3 syntax
 # cython: language_level = 3
 
+import cupy
+import numpy
+
 from pylibcugraph._cugraph_c.types cimport (
     bool_t,
 )
@@ -167,20 +170,6 @@ def pagerank(ResourceHandle resource_handle,
     array([0.11615585, 0.21488841, 0.2988108 , 0.3701449 ], dtype=float32)
     """
 
-    # FIXME: import these modules here for now until a better pattern can be
-    # used for optional imports (perhaps 'import_optional()' from cugraph), or
-    # these are made hard dependencies.
-    try:
-        import cupy
-    except ModuleNotFoundError:
-        raise RuntimeError("pagerank requires the cupy package, which could "
-                           "not be imported")
-    try:
-        import numpy
-    except ModuleNotFoundError:
-        raise RuntimeError("pagerank requires the numpy package, which could "
-                           "not be imported")
-
     cdef cugraph_type_erased_device_array_view_t* \
         initial_guess_vertices_view_ptr = \
             create_cugraph_type_erased_device_array_view_from_py_obj(

python/pylibcugraph/pylibcugraph/personalized_pagerank.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -14,6 +14,9 @@
 # Have cython use python 3 syntax
 # cython: language_level = 3
 
+import cupy
+import numpy
+
 from pylibcugraph._cugraph_c.types cimport (
     bool_t,
 )
@@ -177,20 +180,6 @@ def personalized_pagerank(ResourceHandle resource_handle,
     array([0.00446455, 0.00379487, 0.53607565, 0.45566472 ], dtype=float32)
     """
 
-    # FIXME: import these modules here for now until a better pattern can be
-    # used for optional imports (perhaps 'import_optional()' from cugraph), or
-    # these are made hard dependencies.
-    try:
-        import cupy
-    except ModuleNotFoundError:
-        raise RuntimeError("pagerank requires the cupy package, which could "
-                           "not be imported")
-    try:
-        import numpy
-    except ModuleNotFoundError:
-        raise RuntimeError("pagerank requires the numpy package, which could "
-                           "not be imported")
-
     cdef cugraph_type_erased_device_array_view_t* \
         initial_guess_vertices_view_ptr = \
             create_cugraph_type_erased_device_array_view_from_py_obj(

python/pylibcugraph/pylibcugraph/sssp.pyx

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -14,6 +14,9 @@
 # Have cython use python 3 syntax
 # cython: language_level = 3
 
+import cupy
+import numpy
+
 from pylibcugraph._cugraph_c.types cimport (
     bool_t,
 )
@@ -124,20 +127,6 @@ def sssp(ResourceHandle resource_handle,
     array([-1, -1,  1,  2], dtype=int32)
     """
 
-    # FIXME: import these modules here for now until a better pattern can be
-    # used for optional imports (perhaps 'import_optional()' from cugraph), or
-    # these are made hard dependencies.
-    try:
-        import cupy
-    except ModuleNotFoundError:
-        raise RuntimeError("sssp requires the cupy package, which could not "
-                           "be imported")
-    try:
-        import numpy
-    except ModuleNotFoundError:
-        raise RuntimeError("sssp requires the numpy package, which could not "
-                           "be imported")
-
     if compute_predecessors is False:
         raise ValueError("compute_predecessors must be True for the current "
                          "release.")

python/pylibcugraph/pyproject.toml

@@ -23,6 +23,8 @@ authors = [
 license = { text = "Apache 2.0" }
 requires-python = ">=3.10"
 dependencies = [
+    "cupy-cuda11x>=12.0.0",
+    "numpy>=1.23,<3.0a0",
     "nvidia-cublas",
     "nvidia-curand",
     "nvidia-cusolver",