Support release-candidate builds as a first-class mode

tests/test_rapids-generate-version.py

@@ -0,0 +1,93 @@
+import os
+import subprocess
+from pathlib import Path
+
+import pytest
+
+TOOLS_DIRECTORY = Path(__file__).resolve().parents[1] / "tools"
+
+
+def _generate_version(
+    tmp_path: Path,
+    candidate_version: str | None,
+    source_version: str | None = None,
+) -> subprocess.CompletedProcess[str]:
+    environment = os.environ.copy()
+    environment.pop("RAPIDS_RELEASE_CANDIDATE_SOURCE_VERSION", None)
+    if candidate_version is None:
+        environment.pop("RAPIDS_RELEASE_CANDIDATE_VERSION", None)
+    else:
+        environment["RAPIDS_RELEASE_CANDIDATE_VERSION"] = candidate_version
+    if source_version is not None:
+        environment["RAPIDS_RELEASE_CANDIDATE_SOURCE_VERSION"] = source_version
+    return subprocess.run(
+        [TOOLS_DIRECTORY / "rapids-generate-version"],
+        cwd=tmp_path,
+        env=environment,
+        text=True,
+        capture_output=True,
+        check=False,
+    )
+
+
+def test_release_candidate_version_returns_exact_final_version_without_git_tag(tmp_path):
+    tmp_path.joinpath("VERSION").write_text("26.10.00\n")
+
+    result = _generate_version(tmp_path, "26.10.00")
+
+    assert result.returncode == 0
+    assert result.stdout == "26.10.00"
+    assert result.stderr == ""
+
+
+@pytest.mark.parametrize("candidate_version", ["v26.10.00", "26", "26.10.00.1", "26.10.00rc0"])
+def test_release_candidate_version_rejects_non_final_formats(tmp_path, candidate_version):
+    tmp_path.joinpath("VERSION").write_text("26.10.00\n")
+
+    result = _generate_version(tmp_path, candidate_version)
+
+    assert result.returncode == 1
+    assert "must use a numeric YY.MM or YY.MM.XX format" in result.stderr
+
+
+def test_release_candidate_version_supports_independently_versioned_repository(tmp_path):
+    tmp_path.joinpath("VERSION").write_text("0.3.0\n")
+
+    result = _generate_version(tmp_path, "0.3.0")
+
+    assert result.returncode == 0
+    assert result.stdout == "0.3.0"
+
+
+def test_release_candidate_version_supports_ucxx_two_component_version(tmp_path):
+    tmp_path.joinpath("VERSION").write_text("0.52.00\n")
+
+    result = _generate_version(tmp_path, "0.52")
+
+    assert result.returncode == 0
+    assert result.stdout == "0.52"
+
+
+def test_release_candidate_version_rejects_different_source_major_minor(tmp_path):
+    tmp_path.joinpath("VERSION").write_text("26.12.00a0\n")
+
+    result = _generate_version(tmp_path, "26.10.00")
+
+    assert result.returncode == 1
+    assert "does not match source major/minor '26.12'" in result.stderr
+
+
+def test_release_candidate_version_uses_preserved_source_after_output_truncation(tmp_path):
+    tmp_path.joinpath("VERSION").write_text("")
+
+    result = _generate_version(tmp_path, "26.10.00", source_version="26.10.00")
+
+    assert result.returncode == 0
+    assert result.stdout == "26.10.00"
+
+
+def test_release_candidate_version_requires_version_file(tmp_path):
+    result = _generate_version(tmp_path, "26.10.00")
+
+    assert result.returncode == 1
+    assert "requires a VERSION file" in result.stderr

tests/test_release_candidate_build_type.py

@@ -0,0 +1,44 @@
+import os
+import subprocess
+from pathlib import Path
+
+TOOLS_DIRECTORY = Path(__file__).resolve().parents[1] / "tools"
+
+
+def _environment(**updates: str) -> dict[str, str]:
+    environment = os.environ.copy()
+    environment["PATH"] = f"{TOOLS_DIRECTORY}:{environment['PATH']}"
+    environment.update(updates)
+    return environment
+
+
+def test_release_candidate_is_a_release_build():
+    result = subprocess.run(
+        [TOOLS_DIRECTORY / "rapids-is-release-build"],
+        env=_environment(RAPIDS_BUILD_TYPE="release-candidate", GITHUB_REF="refs/heads/main"),
+        text=True,
+        capture_output=True,
+        check=False,
+    )
+
+    assert result.returncode == 0
+    assert "is release build" in result.stderr
+
+
+def test_release_candidate_rattler_channels_exclude_public_rapids_channels():
+    command = f'source "{TOOLS_DIRECTORY / "rapids-rattler-channel-string"}"; printf "%s\\n" "${{RATTLER_CHANNELS[*]}}"'
+    result = subprocess.run(
+        ["bash", "-c", command],
+        env=_environment(
+            RAPIDS_BUILD_TYPE="release-candidate",
+            GITHUB_REF="refs/heads/main",
+            RAPIDS_CONDA_BLD_OUTPUT_DIR="/tmp/conda-output",
+        ),
+        text=True,
+        capture_output=True,
+        check=False,
+    )
+
+    assert result.returncode == 0
+    assert result.stdout.splitlines()[-1] == "--channel conda-forge"
+    assert "rapidsai" not in result.stdout

tools/rapids-generate-version

@@ -3,6 +3,40 @@
 set -euo pipefail
 export RAPIDS_SCRIPT_NAME="rapids-generate-version"
 
+if [[ -n "${RAPIDS_RELEASE_CANDIDATE_VERSION:-}" ]]; then
+    readonly candidate_version="${RAPIDS_RELEASE_CANDIDATE_VERSION}"
+    readonly candidate_version_regex='^([0-9]+)\.([0-9]+)(\.[0-9]+)?$'
+    if [[ ! "${candidate_version}" =~ ${candidate_version_regex} ]]; then
+        echo "RAPIDS_RELEASE_CANDIDATE_VERSION must use a numeric YY.MM or YY.MM.XX format, got '${candidate_version}'" >&2
+        exit 1
+    fi
+    # A successful Bash '=~' comparison stores each parenthesized capture in
+    # BASH_REMATCH. Save these captures before the source-version match below
+    # replaces them.
+    readonly candidate_major_minor="${BASH_REMATCH[1]}.${BASH_REMATCH[2]}"
+    source_version="${RAPIDS_RELEASE_CANDIDATE_SOURCE_VERSION:-}"
+    if [[ -z "${source_version}" ]]; then
+        if [[ ! -f VERSION ]]; then
+            echo "RAPIDS_RELEASE_CANDIDATE_VERSION requires a VERSION file or RAPIDS_RELEASE_CANDIDATE_SOURCE_VERSION" >&2
+            exit 1
+        fi
+        source_version="$(head -n 1 VERSION)"
+    fi
+    readonly source_version
+    readonly source_version_regex='^([0-9]+)\.([0-9]+)(\.|$)'
+    if [[ ! "${source_version}" =~ ${source_version_regex} ]]; then
+        echo "source version must begin with two numeric components, got '${source_version}'" >&2
+        exit 1
+    fi
+    readonly source_major_minor="${BASH_REMATCH[1]}.${BASH_REMATCH[2]}"
+    if [[ "${source_major_minor}" != "${candidate_major_minor}" ]]; then
+        echo "release-candidate version '${candidate_version}' does not match source major/minor '${source_major_minor}'" >&2
+        exit 1
+    fi
+    echo -n "${candidate_version}"
+    exit 0
+fi
+
 if rapids-is-release-build; then
     dunamai_version=$(python -m dunamai from git --format "{base}")
 else

tools/rapids-github-run-id

@@ -34,12 +34,12 @@ else
       workflow_that_produced_artifacts="pr.yaml"
       workflow_that_produced_artifacts_regex="\\.github/workflows/pr\\.yaml"
       ;;
-    branch|nightly)
+    branch|nightly|release-candidate)
       workflow_that_produced_artifacts="build.yaml"
       workflow_that_produced_artifacts_regex="\\.github/workflows/build\\.yaml"
       ;;
     *)
-      rapids-echo-stderr "RAPIDS_BUILD_TYPE must be one of [branch, nightly, pull-request]"
+      rapids-echo-stderr "RAPIDS_BUILD_TYPE must be one of [branch, nightly, pull-request, release-candidate]"
       exit 1
       ;;
   esac

tools/rapids-is-release-build

@@ -1,14 +1,15 @@
 #!/bin/bash
 # A utility script that examines environment variables provided by
 # GitHub Actions to determine whether the current build is a "release" build.
-# A "release" build occurs when the GITHUB_REF environment variable matches
-# the pattern "refs/tags/vYY.MM.PP".
+# A "release" build occurs when the workflow explicitly specifies a release
+# candidate or when GITHUB_REF matches the pattern "refs/tags/vYY.MM.PP".
 # Example:
 #    if rapids-is-release-build; then echo "hi"; fi
 set -e
 export RAPIDS_SCRIPT_NAME="rapids-is-release-build"
 
-if [[ "${GITHUB_REF}" =~ ^refs/tags/v[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
+if [[ "${RAPIDS_BUILD_TYPE:-}" == "release-candidate" ]] ||
+   [[ "${GITHUB_REF:-}" =~ ^refs/tags/v[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
   rapids-echo-stderr "is release build"
   exit 0
 fi

tools/rapids-prompt-local-repo-config

@@ -7,7 +7,8 @@
 #
 # Variables:
 #
-#   * RAPIDS_BUILD_TYPE   = One of "branch", "nightly", or "pull-request".
+#   * RAPIDS_BUILD_TYPE   = One of "branch", "nightly", "pull-request", or
+#                           "release-candidate".
 #   * RAPIDS_NIGHTLY_DATE = Date in YYYY-MM-DD format, used to organize nightly uploads.
 #                           If not provided, the current system time is used.
 #                           Only used when `RAPIDS_BUILD_TYPE` is "nightly".
@@ -24,7 +25,7 @@ if [ "${CI:-false}" = "false" ]; then
   if [ -z "${RAPIDS_BUILD_TYPE:-}" ]; then
     {
       echo ""
-      read -r -p "Enter workflow type (one of: pull-request|branch|nightly): " RAPIDS_BUILD_TYPE
+      read -r -p "Enter workflow type (one of: pull-request|branch|nightly|release-candidate): " RAPIDS_BUILD_TYPE
       export RAPIDS_BUILD_TYPE
       echo ""
       echo "Suppress this prompt in the future by setting the 'RAPIDS_BUILD_TYPE' environment variable:"

tools/rapids-rattler-channel-string

@@ -35,8 +35,11 @@
 
 RAPIDS_CHANNEL="rapidsai-nightly"
 
-# Replace dev/nightly channels if build is a release build
-if rapids-is-release-build; then
+# Candidate dependencies come only from the frozen local candidate channel and
+# conda-forge. They must not resolve from either public RAPIDS channel.
+if [[ "${RAPIDS_BUILD_TYPE:-}" == "release-candidate" ]]; then
+  RAPIDS_CHANNEL=""
+elif rapids-is-release-build; then
   RAPIDS_CHANNEL="rapidsai"
 fi