rapidsai · rapids-bot · Feb 28, 2023 · Jan 18, 2023 · Jan 19, 2023 · Jan 20, 2023
@@ -52,6 +52,17 @@ jobs:
       branch: ${{ inputs.branch }}
       date: ${{ inputs.date }}
       sha: ${{ inputs.sha }}
+  docs-build:
+    if: github.ref_type == 'branch' && github.event_name == 'push'
+    needs: python-build
+    secrets: inherit
+    uses: rapidsai/shared-action-workflows/.github/workflows/[email protected]
+    with:
+      build_type: branch
+      node_type: "gpu-latest-1"
+      arch: "amd64"
+      container_image: "rapidsai/ci:latest"
+      run_script: "ci/build_docs.sh"
   wheel-build-pylibcugraph:
     secrets: inherit
     uses: rapidsai/shared-action-workflows/.github/workflows/[email protected]

@@ -18,6 +18,7 @@ jobs:
       - conda-notebook-tests
       - conda-python-build
       - conda-python-tests
+      - docs-build
       - wheel-build-pylibcugraph
       - wheel-tests-pylibcugraph
       - wheel-build-cugraph
@@ -62,6 +63,16 @@ jobs:
       arch: "amd64"
       container_image: "rapidsai/ci:latest"
       run_script: "ci/test_notebooks.sh"
+  docs-build:
+    needs: conda-python-build
+    secrets: inherit
+    uses: rapidsai/shared-action-workflows/.github/workflows/[email protected]
+    with:
+      build_type: pull-request
+      node_type: "gpu-latest-1"
+      arch: "amd64"
+      container_image: "rapidsai/ci:latest"
+      run_script: "ci/build_docs.sh"
   wheel-build-pylibcugraph:
     needs: checks
     secrets: inherit

diff --git a/.gitignore b/.gitignore
@@ -93,3 +93,5 @@ python/cugraph/cugraph/tests/dask-worker-space
 
 # Sphinx docs & build artifacts
 docs/cugraph/source/api_docs/api/*
+_html
+_text
@@ -0,0 +1,63 @@
+#!/bin/bash
+# Copyright (c) 2023, NVIDIA CORPORATION.
+
+set -euo pipefail
+
+rapids-logger "Create test conda environment"
+. /opt/conda/etc/profile.d/conda.sh
+
+rapids-dependency-file-generator \
+  --output conda \
+  --file_key docs \
+  --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION}" | tee env.yaml
+
+rapids-mamba-retry env create --force -f env.yaml -n docs
+conda activate docs
+
+rapids-print-env
+
+rapids-logger "Downloading artifacts from previous jobs"
+CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
+PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
+VERSION_NUMBER=$(rapids-get-rapids-version-from-git)
+
+rapids-mamba-retry install \
+  --channel "${CPP_CHANNEL}" \
+  --channel "${PYTHON_CHANNEL}" \
+  libcugraph \
+  pylibcugraph \
+  cugraph \
+  cugraph-pyg \
+  cugraph-service-server \
+  cugraph-service-client \
+  libcugraph_etl
+
+# This command installs `cugraph-dgl` without its dependencies
+# since this package can currently only run in `11.6` CTK environments
+# due to the dependency version specifications in its conda recipe.
+rapids-logger "Install cugraph-dgl"
+set -x
+rapids-mamba-retry install "${PYTHON_CHANNEL}/linux-64/cugraph-dgl-*.tar.bz2"
+set +x
+
+
+# Build Doxygen docs
+rapids-logger "Build Doxygen docs"
+pushd cpp/doxygen
+doxygen Doxyfile
+popd
+
+# Build Python docs
+rapids-logger "Build Sphinx docs"
+pushd docs/cugraph
+sphinx-build -b dirhtml source _html
+sphinx-build -b text source _text
+popd
+
+
+if [[ "${RAPIDS_BUILD_TYPE}" == "branch" ]]; then
+  rapids-logger "Upload Docs to S3"
+  aws s3 sync --delete docs/cugraph/_html "s3://rapidsai-docs/cugraph/${VERSION_NUMBER}/html"
+  aws s3 sync --delete docs/cugraph/_text "s3://rapidsai-docs/cugraph/${VERSION_NUMBER}/txt"
+  aws s3 sync --delete cpp/doxygen/html "s3://rapidsai-docs/libcugraph/${VERSION_NUMBER}/html"
+fi
@@ -22,6 +22,7 @@ dependencies:
 - doxygen
 - gcc_linux-64=9.*
 - gmock=1.10.0
+- graphviz
 - gtest=1.10.0
 - ipython
 - libcudf=23.04.*
@@ -54,9 +55,10 @@ dependencies:
 - scikit-build>=0.13.1
 - scikit-learn>=0.23.1
 - scipy
-- sphinx
 - sphinx-copybutton
 - sphinx-markdown-tables
+- sphinx<6
+- sphinx_rtd_theme
 - sphinxcontrib-websupport
 - ucx-proc=*=gpu
 - ucx-py=0.31.*

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -11,7 +11,7 @@ files:
       - common_python_test
       - cpp_build
       - cudatoolkit
-      - doc
+      - docs
       - python_build
       - test_notebook
       - test_python
@@ -20,6 +20,12 @@ files:
     includes:
       - checks
       - py_version
+  docs:
+    output: none
+    includes:
+      - cudatoolkit
+      - docs
+      - py_version
   test_cpp:
     output: none
     includes:
@@ -113,6 +119,22 @@ dependencies:
               cuda: "11.8"
             packages:
               - nvcc_linux-aarch64=11.8
+  docs:
+    common:
+      - output_types: [conda]
+        packages:
+          - doxygen
+          - graphviz
+          - ipython
+          - nbsphinx
+          - numpydoc
+          - pydata-sphinx-theme
+          - recommonmark
+          - sphinx_rtd_theme
+          - sphinx-copybutton
+          - sphinx-markdown-tables
+          - sphinx<6
+          - sphinxcontrib-websupport
   py_version:
     specific:
       - output_types: [conda]

@@ -28,7 +28,7 @@ Katz Centrality (MG)
    cugraph.dask.centrality.katz_centrality.katz_centrality
 
 Degree Centrality
----------------
+-----------------
 .. autosummary::
    :toctree: api/
 

@@ -10,6 +10,5 @@ cugraph-pyg
    :toctree: api/
 
    cugraph_pyg.data.cugraph_store.EXPERIMENTAL__CuGraphStore
-   cugraph_pyg.loader.dispatch.call_cugraph_algorithm
    cugraph_pyg.sampler.cugraph_sampler.EXPERIMENTAL__CuGraphSampler
 
@@ -1,5 +1,5 @@
 Python API reference
-=============
+====================
 
 This page provides a list of all publicly accessible modules, methods and classes through
 ``cugraph.*`` namespace.

@@ -1,6 +1,3 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-#
 # Copyright (c) 2018-2023, NVIDIA CORPORATION.
 #
 # pygdf documentation build configuration file, created by
@@ -27,8 +24,6 @@
 # is relative to the documentation root, use os.path.abspath to make it
 # absolute, like shown here.
 sys.path.insert(0, os.path.abspath('sphinxext'))
-sys.path.insert(0, os.path.abspath('../../python'))
-sys.path.insert(0, os.path.abspath('../..'))
 
 from github_link import make_linkcode_resolve # noqa
 
@@ -112,18 +107,6 @@
 
 html_theme = 'pydata_sphinx_theme'
 
-# on_rtd is whether we are on readthedocs.org
-on_rtd = os.environ.get('READTHEDOCS', None) == 'True'
-
-if not on_rtd:
-    # only import and set the theme if we're building docs locally
-    # otherwise, readthedocs.org uses their theme by default,
-    # so no need to specify it
-    import pydata_sphinx_theme
-    import sphinx_rtd_theme
-    html_theme = 'pydata_sphinx_theme'
-    html_theme_path = sphinx_rtd_theme.get_html_theme_path()
-
 # If true, `todo` and `todoList` produce output, else they produce nothing.
 todo_include_todos = False
 

@@ -83,6 +83,7 @@ def __init__(
             information this can be ``torch.int32`` or ``torch.int64``
             for PyTorch.
             Defaults to ``torch.int64`` if pytorch is installed
+
          Examples
          --------
          The following example uses `CuGraphStorage` :

@@ -13,21 +13,22 @@
 from __future__ import annotations
 import os
 import shutil
-import torch
 import cugraph_dgl
 import cupy as cp
 import cudf
 from cugraph.experimental import BulkSampler
 from dask.distributed import default_client, Event
-import dgl
-from dgl.dataloading import WorkerInitWrapper, create_tensorized_dataset
 from cugraph_dgl.dataloading import (
     HomogenousBulkSamplerDataset,
     HetrogenousBulkSamplerDataset,
 )
 from cugraph_dgl.dataloading.utils.extract_graph_helpers import (
     create_cugraph_graph_from_edges_dict,
 )
+from cugraph.utilities.utils import import_optional
+
+dgl = import_optional("dgl")
+torch = import_optional("torch")
 
 
 class DataLoader(torch.utils.data.DataLoader):
@@ -128,7 +129,9 @@ def __init__(
         self.shuffle = shuffle
         self.drop_last = drop_last
         self.graph_sampler = graph_sampler
-        worker_init_fn = WorkerInitWrapper(kwargs.get("worker_init_fn", None))
+        worker_init_fn = dgl.dataloading.WorkerInitWrapper(
+            kwargs.get("worker_init_fn", None)
+        )
         self.other_storages = {}
         self.epoch_number = 0
         self._batch_size = batch_size
@@ -138,7 +141,7 @@ def __init__(
 
         indices = _dgl_idx_to_cugraph_idx(indices, graph)
 
-        self.tensorized_indices_ds = create_tensorized_dataset(
+        self.tensorized_indices_ds = dgl.dataloading.create_tensorized_dataset(
             indices, batch_size, drop_last, use_ddp, ddp_seed, shuffle
         )
         if len(graph.ntypes) <= 1:

@@ -20,9 +20,10 @@
     create_heterogeneous_sampled_graphs_from_dataframe,
 )
 
-# TODO: Make optional imports
-import torch
-import dgl
+from cugraph.utilities.utils import import_optional
+
+dgl = import_optional("dgl")
+torch = import_optional("torch")
 
 
 # Todo: maybe should switch to __iter__

@@ -12,16 +12,16 @@
 # limitations under the License.
 from __future__ import annotations
 from typing import Tuple, Dict
-from torch.utils.dlpack import from_dlpack
 from collections import defaultdict
 import cudf
-import torch
-import dgl
+from cugraph.utilities.utils import import_optional
+
+dgl = import_optional("dgl")
+torch = import_optional("torch")
 
 
 def cast_to_tensor(ser: cudf.Series):
-    # TODO: Maybe use torch.as_tensor
-    return from_dlpack(ser.values.toDlpack())
+    return torch.as_tensor(ser.values, device="cuda")
 
 
 def create_homogeneous_sampled_graphs_from_dataframe(

@@ -13,6 +13,7 @@
 """Torch Module for graph attention network layer using the aggregation
 primitives in cugraph-ops"""
 # pylint: disable=no-member, arguments-differ, invalid-name, too-many-arguments
+from __future__ import annotations
 from typing import Optional
 
 from cugraph.utilities.utils import import_optional

@@ -13,6 +13,7 @@
 """Torch Module for Relational graph convolution layer using the aggregation
 primitives in cugraph-ops"""
 # pylint: disable=no-member, arguments-differ, invalid-name, too-many-arguments
+from __future__ import annotations
 import math
 from typing import Optional
 

@@ -13,6 +13,7 @@
 """Torch Module for GraphSAGE layer using the aggregation primitives in
 cugraph-ops"""
 # pylint: disable=no-member, arguments-differ, invalid-name, too-many-arguments
+from __future__ import annotations
 from typing import Optional
 
 from cugraph.utilities.utils import import_optional

diff --git a/python/cugraph/_custom_build/backend.py b/python/cugraph/_custom_build/backend.py
@@ -22,7 +22,7 @@ def wrapper(config_settings=None):
         cuda_suffix = os.getenv("RAPIDS_PY_WHEEL_CUDA_SUFFIX", default="")
         append_list = [
             f"rmm{cuda_suffix}==23.4.*",
-            f"raft-dask{cuda_suffix}==23.4.*",
+            f"pylibraft{cuda_suffix}==23.4.*",
             f"pylibcugraph{cuda_suffix}==23.4.*",
         ]
         return orig_list + append_list

@@ -1,4 +1,4 @@
-# Copyright (c) 2019-2022, NVIDIA CORPORATION.
+# Copyright (c) 2019-2023, 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
@@ -77,7 +77,9 @@ def pagerank(
     increases when the tolerance descreases and/or alpha increases toward the
     limiting value of 1. The user is free to use default values or to provide
     inputs for the initial guess, tolerance and maximum number of iterations.
-    Parameters. All edges will have an edge_attr value of 1.0 if not provided.
+    All edges will have an edge_attr value of 1.0 if not provided.
+
+    Parameters
     ----------
     G : cugraph.Graph or networkx.Graph
         cuGraph graph descriptor, should contain the connectivity information
@@ -158,6 +160,7 @@ def pagerank(
             Contains the vertex identifiers
         df['pagerank'] : cudf.Series
             Contains the PageRank score
+
     Examples
     --------
     >>> from cugraph.experimental.datasets import karate