diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index d1974a7c5..59e188881 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -35,7 +35,7 @@ jobs:
       date: ${{ inputs.date }}
       sha: ${{ inputs.sha }}
   docs-build:
-    if: ${{ startsWith(github.ref, 'refs/heads/branch-') }}
+    if: github.ref_type == 'branch' && github.event_name == 'push'
     needs: [conda-python-build]
     secrets: inherit
     uses: rapidsai/shared-action-workflows/.github/workflows/custom-job.yaml@branch-23.04
diff --git a/dependencies.yaml b/dependencies.yaml
index bb99ea21a..fbbbf05d1 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -100,7 +100,7 @@ dependencies:
           - numba>=0.54
           - numpy>=1.18.0
           - pandas>=1.0
-          - pynvml>=11.0.0
+          - pynvml>=11.0.0,<11.5
           - zict>=0.1.3
   test_python:
     common:
diff --git a/pyproject.toml b/pyproject.toml
index 6d3828673..cbd6c25f8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,7 +21,7 @@ requires-python = ">=3.8"
 dependencies = [
     "dask >=2023.1.1",
     "distributed >=2023.1.1",
-    "pynvml >=11.0.0",
+    "pynvml >=11.0.0,<11.5",
     "numpy >=1.18.0",
     "numba >=0.54",
     "pandas >=1.0",