From d357df99d83fd832a9e3b2662744d432f54dfb3d Mon Sep 17 00:00:00 2001
From: Jaya Venkatesh <jjayabaskar@nvidia.com>
Date: Mon, 23 Mar 2026 20:38:47 -0700
Subject: [PATCH 1/2] added run-notebooks file key for cugraph

Signed-off-by: Jaya Venkatesh <jjayabaskar@nvidia.com>
---
 context/notebooks.sh | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/context/notebooks.sh b/context/notebooks.sh
index ca7462bb..5c00d0bd 100755
--- a/context/notebooks.sh
+++ b/context/notebooks.sh
@@ -29,11 +29,20 @@ for REPO in "${NOTEBOOK_REPOS[@]}"; do
         cp -rL "$SOURCE" "$DESTINATION"
     fi
 
-    if [ -f "$REPO/dependencies.yaml" ] && yq -e '.files.test_notebooks' "$REPO/dependencies.yaml" >/dev/null; then
-        echo "Running dfg on $REPO"
+    if [ -f "$REPO/dependencies.yaml" ]; then
+        # Prefer run_notebooks (runtime-only deps) over test_notebooks
+        # (which may include testing libraries like pytest).
+        if yq -e '.files.run_notebooks' "$REPO/dependencies.yaml" >/dev/null 2>&1; then
+            FILE_KEY="run_notebooks"
+        elif yq -e '.files.test_notebooks' "$REPO/dependencies.yaml" >/dev/null 2>&1; then
+            FILE_KEY="test_notebooks"
+        else
+            continue
+        fi
+        echo "Running dfg on $REPO (file-key: $FILE_KEY)"
         rapids-dependency-file-generator \
             --config "$REPO/dependencies.yaml" \
-            --file-key test_notebooks \
+            --file-key "$FILE_KEY" \
             --matrix "cuda=${CUDA_VER%.*};arch=$(arch);py=${PYTHON_VER}" \
             --output conda >"/dependencies/${REPO}_notebooks_tests_dependencies.yaml"
     fi

From f09eb26506bf05bd6653a0c26611e01ab7c664b8 Mon Sep 17 00:00:00 2001
From: Jaya Venkatesh <jjayabaskar@nvidia.com>
Date: Tue, 24 Mar 2026 10:19:57 -0700
Subject: [PATCH 2/2] update file-keys with hard requirements

Signed-off-by: Jaya Venkatesh <jjayabaskar@nvidia.com>
---
 context/notebooks.sh | 26 ++++++++++----------------
 1 file changed, 10 insertions(+), 16 deletions(-)

diff --git a/context/notebooks.sh b/context/notebooks.sh
index 5c00d0bd..ea99e71e 100755
--- a/context/notebooks.sh
+++ b/context/notebooks.sh
@@ -29,23 +29,17 @@ for REPO in "${NOTEBOOK_REPOS[@]}"; do
         cp -rL "$SOURCE" "$DESTINATION"
     fi
 
-    if [ -f "$REPO/dependencies.yaml" ]; then
-        # Prefer run_notebooks (runtime-only deps) over test_notebooks
-        # (which may include testing libraries like pytest).
-        if yq -e '.files.run_notebooks' "$REPO/dependencies.yaml" >/dev/null 2>&1; then
-            FILE_KEY="run_notebooks"
-        elif yq -e '.files.test_notebooks' "$REPO/dependencies.yaml" >/dev/null 2>&1; then
-            FILE_KEY="test_notebooks"
-        else
-            continue
-        fi
-        echo "Running dfg on $REPO (file-key: $FILE_KEY)"
-        rapids-dependency-file-generator \
-            --config "$REPO/dependencies.yaml" \
-            --file-key "$FILE_KEY" \
-            --matrix "cuda=${CUDA_VER%.*};arch=$(arch);py=${PYTHON_VER}" \
-            --output conda >"/dependencies/${REPO}_notebooks_tests_dependencies.yaml"
+    if [[ "${REPO}" == "cugraph" ]]; then
+        FILE_KEY="run_notebooks"
+    else
+        FILE_KEY="test_notebooks"
     fi
+    echo "Running dfg on $REPO (file-key: $FILE_KEY)"
+    rapids-dependency-file-generator \
+        --config "$REPO/dependencies.yaml" \
+        --file-key "$FILE_KEY" \
+        --matrix "cuda=${CUDA_VER%.*};arch=$(arch);py=${PYTHON_VER}" \
+        --output conda >"/dependencies/${REPO}_notebooks_tests_dependencies.yaml"
 done
 
 pushd "/dependencies"