Remove testing libraries in the base and notebook images

context/notebooks.sh

@@ -1,5 +1,5 @@
 #!/usr/bin/env bash
-# Copyright (c) 2023-2025, NVIDIA CORPORATION.
+# Copyright (c) 2023-2026, NVIDIA CORPORATION.
 
 # Clones repos with notebooks & compiles notebook test dependencies
 # Requires environment variables:
@@ -29,14 +29,17 @@ 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"
-        rapids-dependency-file-generator \
-            --config "$REPO/dependencies.yaml" \
-            --file-key test_notebooks \
-            --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"