From 16e614c7f2b58f410e2aecee741ab6bc6c620f11 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Fri, 15 Nov 2024 12:23:53 -0600
Subject: [PATCH] remove versioning workaround for nightlies (#69)

Within this release, packages here moved from
https://github.com/rapidsai/cugraph and
https://github.com/rapidsai/wholegraph to this repo.

To avoid conflicts between nightlies produced from the different
sources, CI scripts here were hard-coding the version for CI artifacts
to `24.12.00a1000`. That should be removed now that this is the only
repo packages are being published from (should have been removed in #66
, sorry for missing it).

---------

Co-authored-by: Ray Douglass <ray@raydouglass.com>
---
 ci/build_cpp.sh              |  6 +-----
 ci/build_python.sh           |  6 +-----
 ci/release/update-version.sh |  5 +++++
 ci/test_notebooks.sh         |  5 +----
 ci/test_python.sh            | 19 +++++--------------
 5 files changed, 13 insertions(+), 28 deletions(-)

diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh
index ed464bd..7202637 100755
--- a/ci/build_cpp.sh
+++ b/ci/build_cpp.sh
@@ -13,11 +13,7 @@ export CMAKE_GENERATOR=Ninja
 
 rapids-print-env
 
-# TODO: revert this once we start publishing nightly packages
-#       from the 'cugraph-gnn' repo and stop publishing them from
-#       the 'cugraph' / 'wholegraph' repos
-#version=$(rapids-generate-version)
-version="24.12.00a1000"
+version=$(rapids-generate-version)
 
 rapids-logger "Begin cpp build"
 
diff --git a/ci/build_python.sh b/ci/build_python.sh
index c74c7e2..e3a2c02 100755
--- a/ci/build_python.sh
+++ b/ci/build_python.sh
@@ -15,11 +15,7 @@ rapids-print-env
 
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 
-# TODO: revert this once we start publishing nightly packages
-#       from the 'cugraph-gnn' repo and stop publishing them from
-#       the 'cugraph' / 'wholegraph' repos
-# rapids-generate-version > ./VERSION
-echo "24.12.00a1000" > ./VERSION
+rapids-generate-version > ./VERSION
 
 sccache --zero-stats
 
diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
index c5cdcb0..88a8ea9 100755
--- a/ci/release/update-version.sh
+++ b/ci/release/update-version.sh
@@ -84,3 +84,8 @@ for DEP in "${DEPENDENCIES[@]}"; do
 done
 
 sed_runner "s/\(PROJECT_NUMBER[[:space:]]*\)=.*/\1= ${NEXT_SHORT_TAG}/" cpp/Doxyfile
+
+# CI files
+for FILE in .github/workflows/*.yaml; do
+  sed_runner "/shared-workflows/ s/@.*/@branch-${NEXT_SHORT_TAG}/g" "${FILE}"
+done
diff --git a/ci/test_notebooks.sh b/ci/test_notebooks.sh
index 1a18e60..00c63af 100755
--- a/ci/test_notebooks.sh
+++ b/ci/test_notebooks.sh
@@ -26,14 +26,11 @@ rapids-logger "Downloading artifacts from previous jobs"
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
 
-# TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
-#       from the 'cugraph-gnn' repo and stop publishing them from
-#       the 'cugraph' / 'wholegraph' repos
 rapids-mamba-retry install \
   --channel "${CPP_CHANNEL}" \
   --channel "${PYTHON_CHANNEL}" \
   --channel dglteam/label/th23_cu118 \
-  "cugraph-dgl=${RAPIDS_VERSION},>=24.12.00a1000"
+  "cugraph-dgl=${RAPIDS_VERSION}"
 
 NBTEST="$(realpath "$(dirname "$0")/utils/nbtest.sh")"
 NOTEBOOK_LIST="$(realpath "$(dirname "$0")/notebook_list.py")"
diff --git a/ci/test_python.sh b/ci/test_python.sh
index 79a90d2..9c02a48 100755
--- a/ci/test_python.sh
+++ b/ci/test_python.sh
@@ -64,9 +64,6 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
   fi
 
 
-  # TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
-  #       from the 'cugraph-gnn' repo and stop publishing them from
-  #       the 'cugraph' / 'wholegraph' repos
   rapids-mamba-retry install \
     --channel "${CPP_CHANNEL}" \
     --channel "${PYTHON_CHANNEL}" \
@@ -74,8 +71,8 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
     --channel conda-forge \
     --channel "${DGL_CHANNEL}" \
     --channel nvidia \
-    "pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
-    "cugraph-dgl=${RAPIDS_VERSION},>=24.12.00a1000" \
+    "pylibwholegraph=${RAPIDS_VERSION}" \
+    "cugraph-dgl=${RAPIDS_VERSION}" \
     'pytorch::pytorch>=2.3,<2.4' \
     "ogb"
 
@@ -108,15 +105,12 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
   conda activate test_cugraph_pyg
   set -u
 
-  # TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
-  #       from the 'cugraph-gnn' repo and stop publishing them from
-  #       the 'cugraph' / 'wholegraph' repos
   rapids-mamba-retry install \
     --channel "${CPP_CHANNEL}" \
     --channel "${PYTHON_CHANNEL}" \
     --channel pytorch \
-    "pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
-    "cugraph-pyg=${RAPIDS_VERSION},>=24.12.00a1000" \
+    "pylibwholegraph=${RAPIDS_VERSION}" \
+    "cugraph-pyg=${RAPIDS_VERSION}" \
     'pytorch::pytorch>=2.3,<2.4' \
     'ogb'
 
@@ -149,15 +143,12 @@ if [[ "${RUNNER_ARCH}" != "ARM64" ]]; then
   conda activate test_pylibwholegraph
   set -u
 
-  # TODO: remove the '>=24.12.00a1000' once we start publishing nightly packages
-  #       from the 'cugraph-gnn' repo and stop publishing them from
-  #       the 'cugraph' / 'wholegraph' repos
   rapids-mamba-retry install \
     --channel "${CPP_CHANNEL}" \
     --channel "${PYTHON_CHANNEL}" \
     --channel pytorch \
     'mkl<2024.1.0' \
-    "pylibwholegraph=${RAPIDS_VERSION},>=24.12.00a1000" \
+    "pylibwholegraph=${RAPIDS_VERSION}" \
     'pytorch::pytorch>=2.3,<2.4' \
     'pytest-forked' \
     'ogb'