Docs improvements

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

@@ -5,7 +5,7 @@ channels:
 - rapidsai
 - conda-forge
 dependencies:
-- breathe
+- breathe>=4.35.0
 - c-compiler
 - clang-tools=20.1.4
 - clang==20.1.4
@@ -34,7 +34,6 @@ dependencies:
 - numpy>=1.23,<3.0
 - numpydoc
 - pre-commit
-- pydata-sphinx-theme
 - pylibraft==26.6.*,>=0.0.0a0
 - pytest
 - pytest-cov
@@ -46,8 +45,10 @@ dependencies:
 - scikit-build-core>=0.11.0
 - scikit-learn
 - scipy
-- sphinx
 - sphinx-copybutton
 - sphinx-markdown-tables
+- sphinx>=8.0.0
 - sysroot_linux-aarch64==2.28
+- pip:
+  - nvidia-sphinx-theme
 name: all_cuda-129_arch-aarch64

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

@@ -5,7 +5,7 @@ channels:
 - rapidsai
 - conda-forge
 dependencies:
-- breathe
+- breathe>=4.35.0
 - c-compiler
 - clang-tools=20.1.4
 - clang==20.1.4
@@ -34,7 +34,6 @@ dependencies:
 - numpy>=1.23,<3.0
 - numpydoc
 - pre-commit
-- pydata-sphinx-theme
 - pylibraft==26.6.*,>=0.0.0a0
 - pytest
 - pytest-cov
@@ -46,8 +45,10 @@ dependencies:
 - scikit-build-core>=0.11.0
 - scikit-learn
 - scipy
-- sphinx
 - sphinx-copybutton
 - sphinx-markdown-tables
+- sphinx>=8.0.0
 - sysroot_linux-64==2.28
+- pip:
+  - nvidia-sphinx-theme
 name: all_cuda-129_arch-x86_64

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

@@ -5,7 +5,7 @@ channels:
 - rapidsai
 - conda-forge
 dependencies:
-- breathe
+- breathe>=4.35.0
 - c-compiler
 - clang-tools=20.1.4
 - clang==20.1.4
@@ -34,7 +34,6 @@ dependencies:
 - numpy>=1.23,<3.0
 - numpydoc
 - pre-commit
-- pydata-sphinx-theme
 - pylibraft==26.6.*,>=0.0.0a0
 - pytest
 - pytest-cov
@@ -46,8 +45,10 @@ dependencies:
 - scikit-build-core>=0.11.0
 - scikit-learn
 - scipy
-- sphinx
 - sphinx-copybutton
 - sphinx-markdown-tables
+- sphinx>=8.0.0
 - sysroot_linux-aarch64==2.28
+- pip:
+  - nvidia-sphinx-theme
 name: all_cuda-131_arch-aarch64

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

@@ -5,7 +5,7 @@ channels:
 - rapidsai
 - conda-forge
 dependencies:
-- breathe
+- breathe>=4.35.0
 - c-compiler
 - clang-tools=20.1.4
 - clang==20.1.4
@@ -34,7 +34,6 @@ dependencies:
 - numpy>=1.23,<3.0
 - numpydoc
 - pre-commit
-- pydata-sphinx-theme
 - pylibraft==26.6.*,>=0.0.0a0
 - pytest
 - pytest-cov
@@ -46,8 +45,10 @@ dependencies:
 - scikit-build-core>=0.11.0
 - scikit-learn
 - scipy
-- sphinx
 - sphinx-copybutton
 - sphinx-markdown-tables
+- sphinx>=8.0.0
 - sysroot_linux-64==2.28
+- pip:
+  - nvidia-sphinx-theme
 name: all_cuda-131_arch-x86_64

dependencies.yaml

@@ -382,16 +382,17 @@ dependencies:
     common:
       - output_types: [conda]
         packages:
-          - breathe
+          - breathe>=4.35.0
           - doxygen>=1.8.20
           - graphviz
           - ipython
           - numpydoc
-          - pydata-sphinx-theme
           - recommonmark
+          - sphinx>=8.0.0
           - sphinx-copybutton
           - sphinx-markdown-tables
-          - sphinx
+          - pip:
+              - nvidia-sphinx-theme
   py_version:
     specific:
       - output_types: conda

docs/source/_static/collapse_overloads.js

@@ -0,0 +1,67 @@
+document.addEventListener("DOMContentLoaded", () => {
+    const toc = document.querySelector(".bd-toc-nav");
+    if (!toc) return;
+
+    // Get all TOC links
+    const links = toc.querySelectorAll("a");
+
+    const seen = new Set();
+    links.forEach(link => {
+      let text = link.textContent.trim();
+      let norm = text.replace(/\(\)$/, ""); // strip trailing ()
+
+      if (seen.has(norm)) {
+        // hide duplicate
+        link.parentElement.style.display = "none";
+      } else {
+        seen.add(norm);
+      }
+    });
+  });
+
+document.addEventListener("DOMContentLoaded", () => {
+  const toc = document.querySelector(".toctree-wrapper");
+  if (!toc) return;
+
+  const leaf_traversal_fn = (leaf) => {
+    try {
+      const links = leaf.querySelectorAll("a");
+      if (!links) return;
+      const seen = new Set();
+      links.forEach(link => {
+        let text = link.textContent.trim();
+        let norm = text.replace(/\(\)$/, ""); // strip trailing ()
+
+        if (seen.has(norm)) {
+          // hide duplicate
+          link.parentElement.style.display = "none";
+        } else {
+          seen.add(norm);
+        }
+      });
+    } catch (error) {
+      console.error(error);
+    }
+  };
+  queue = [toc.querySelector("li")];
+  while (queue.length > 0) {
+    const tree = queue.shift();
+    try {
+      if (tree.childElementCount > 1 &&
+          tree.firstChild.hasAttribute("href") &&
+          tree.firstChild.attributes["href"].value.includes("#")) {
+        leaf_traversal_fn(tree.childNodes[1]);
+      } else {
+        const child = tree.querySelector("li");
+        if (child) {
+          queue.push(child);
+        }
+      }
+    } catch (error) {
+      console.error(error);
+    }
+    if (tree.nextElementSibling) {
+      queue.push(tree.nextElementSibling);
+    }
+  }
+  });

docs/source/_static/references.css

@@ -1,4 +1,15 @@
 
+/* Constrain navbar logo size and add spacing */
+.navbar-brand.logo .logo__image {
+  height: 32px !important;
+  width: auto !important;
+  margin-right: 0.75rem;
+}
+
+.navbar-brand.logo {
+  padding: 0.5rem 0;
+}
+
 /* Fix references to not look like parameters */
 dl.citation > dt.label {
   display: unset !important;
@@ -21,8 +32,3 @@ dl.citation > dt.label > span::before {
 dl.citation > dt.label > span::after {
   content: "]";
 }
-
-/* Make navbar logo smaller */
-.logo__image {
-  height: 32px !important;
-}

docs/source/_templates/navbar-logo.html

@@ -0,0 +1,4 @@
+<a class="navbar-brand logo" href="{{ pathto(root_doc) }}">
+  <img src="{{ logo_url }}" class="logo__image" alt="{{ docstitle }} - Home"/>
+  <p class="title logo__title">{{ docstitle }}</p>
+</a>

docs/source/conf.py

@@ -103,18 +103,16 @@
 # a list of builtin themes.
 #
 
-html_theme = "pydata_sphinx_theme"
-html_logo = "../../img/rapids_logo.png"
+html_theme = "nvidia_sphinx_theme"
+html_logo = "../../img/logo-purple-bg-white-text.svg"
+html_favicon = "../../img/icon-512x512.png"
 
 
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the
 # documentation.
 #
 html_theme_options = {
-    "logo": {
-        "text": "RAFT",
-    },
     "external_links": [],
     # https://github.com/pydata/pydata-sphinx-theme/issues/1220
     "icon_links": [],
@@ -206,6 +204,7 @@ def setup(app):
     app.add_js_file(
         "https://docs.rapids.ai/assets/js/custom.js", loading_method="defer"
     )
+    app.add_js_file("collapse_overloads.js")
 
 
 # The following is used by sphinx.ext.linkcode to provide links to github