Skip to content
19 changes: 11 additions & 8 deletions context/notebooks.sh
Original file line number Diff line number Diff line change
@@ -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:
Expand Down Expand Up @@ -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"
Expand Down
Loading