ci: use pixi to build docs

Author: cpcloud

.github/workflows/docs-build.yaml

@@ -3,28 +3,6 @@
 
 on:
   workflow_call:
-    inputs:
-      build_type:
-        required: true
-        type: string
-      branch:
-        type: string
-      date:
-        type: string
-      sha:
-        type: string
-      repo:
-        type: string
-      node_type:
-        type: string
-        default: "cpu4"
-      script:
-        type: string
-        default: "ci/build_docs.sh"
-
-defaults:
-  run:
-    shell: bash
 
 permissions:
   actions: read
@@ -43,39 +21,16 @@ permissions:
 
 jobs:
   build:
-    strategy:
-      fail-fast: false
-    runs-on: "linux-amd64-cpu4"
-    env:
-      RAPIDS_ARTIFACTS_DIR: ${{ github.workspace }}/artifacts
-    container:
-      image: rapidsai/ci-conda:cuda13.0.0-ubuntu24.04-py3.13
-      env:
-        RAPIDS_BUILD_TYPE: ${{ inputs.build_type }}
+    runs-on: ubuntu-latest
     timeout-minutes: 60
     steps:
       - uses: actions/checkout@v4
+      - name: Setup pixi
+        uses: prefix-dev/setup-pixi@fef5c9568ca6c4ff7707bf840ab0692ba3f08293
         with:
-          repository: ${{ inputs.repo }}
-          ref: ${{ inputs.sha }}
-          fetch-depth: 0
-      - uses: actions/download-artifact@v4
-        with:
-          name: conda-repo-py3.13-amd64
-          path: conda-repo
-      - name: Display structure of downloaded files
-        run: ls -R
-      - name: Standardize repository information
-        run: |
-          echo "RAPIDS_REPOSITORY=${{ inputs.repo || github.repository }}" >> "${GITHUB_ENV}"
-          echo "RAPIDS_SHA=$(git rev-parse HEAD)" >> "${GITHUB_ENV}"
-          echo "RAPIDS_REF_NAME=${{ inputs.branch || github.ref_name }}" >> "${GITHUB_ENV}"
-          echo "RAPIDS_NIGHTLY_DATE=${{ inputs.date }}" >> "${GITHUB_ENV}"
-      - name: Setup proxy cache
-        uses: nv-gha-runners/setup-proxy-cache@main
-        continue-on-error: true
+          environments: docs
       - name: Docs build
-        run: ${{ inputs.script }}
+        run: pixi run -e docs build-docs
         env:
           GH_TOKEN: ${{ github.token }}
       - name: Upload docs

.github/workflows/pr.yaml

@@ -71,7 +71,6 @@ jobs:
       matrix: ${{ needs.compute-matrix.outputs.BUILD_MATRIX }}
   test-conda:
     needs:
-      - build-conda
       - compute-matrix
     uses: ./.github/workflows/conda-python-tests.yaml
     with:
@@ -81,7 +80,6 @@ jobs:
       matrix: ${{ needs.compute-matrix.outputs.TEST_MATRIX }}
   test-simulator:
     needs:
-      - build-conda
       - compute-matrix
     uses: ./.github/workflows/simulator-test.yaml
     with:
@@ -133,11 +131,7 @@ jobs:
       # TODO: Enable for CUDA 13 when a supporting version of cuDF is available
       matrix_filter: map(select(.ARCH == "amd64" and (.CUDA_VER | split(".") | .[0] | tonumber == 12))) | group_by(.CUDA_VER|split(".")|map(tonumber)|.[0]) | map(max_by([(.PY_VER|split(".")|map(tonumber)), (.CUDA_VER|split(".")|map(tonumber))]))
   build-docs:
-    needs:
-      - build-conda
     uses: ./.github/workflows/docs-build.yaml
-    with:
-      build_type: pull-request
   coverage-report:
     needs:
       - build-wheels

.github/workflows/publish.yaml

@@ -67,11 +67,7 @@ jobs:
         with:
           password: ${{ secrets.RAPIDSAI_PYPI_TOKEN }}
   build-docs:
-    needs:
-      - build-conda
     uses: ./.github/workflows/docs-build.yaml
-    with:
-      build_type: release
   deploy-docs:
     needs: build-docs
     permissions:

ci/build_docs.sh

@@ -4,50 +4,15 @@
 
 set -euo pipefail
 
-. /opt/conda/etc/profile.d/conda.sh
-
-rapids-logger "Install docs dependencies"
-# TODO: Replace with rapids-dependency-file-generator
-DEPENDENCIES=(
-    "make"
-    "psutil"
-    "sphinx"
-    "sphinx_rtd_theme"
-    "numpydoc"
-    "python=${RAPIDS_PY_VERSION}"
-    "numba-cuda"
-)
-rapids-mamba-retry create \
-    -n docs \
-    --strict-channel-priority \
-    --channel "`pwd`/conda-repo" \
-    --channel conda-forge \
-    "${DEPENDENCIES[@]}"
-
-# Temporarily allow unbound variables for conda activation.
-set +u
-conda activate docs
-set -u
-
-pip install nvidia-sphinx-theme
-
-rapids-print-env
-
-# Change out of the root of the repo early so the source tree in the repo
-# doesn't confuse things
-pushd docs
-
 rapids-logger "Show Numba system info"
-python -m numba --sysinfo
+pixi run -e docs python -m numba --sysinfo
 
 EXITCODE=0
 trap "EXITCODE=1" ERR
 set +e
 
 rapids-logger "Build docs"
-make html
-
-popd
+pixi run -e docs make -C docs html
 
 rapids-logger "Test script exiting with value: $EXITCODE"
 exit ${EXITCODE}

docs/Makefile

@@ -6,7 +6,7 @@
 
 # You can set these variables from the command line, and also
 # from the environment for the first two.
-SPHINXOPTS    ?= -W
+SPHINXOPTS    ?= -W -j auto
 SPHINXBUILD   ?= sphinx-build
 SOURCEDIR     = source
 BUILDDIR      = build