From bef276f8f7a8c9e8ef97e060ae299d4bebf79873 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Mon, 22 Jun 2026 09:45:04 -0500
Subject: [PATCH 1/2] wheels-publish: upload sdists too

---
 .github/workflows/wheels-publish.yaml | 30 +++++----------------------
 1 file changed, 5 insertions(+), 25 deletions(-)

diff --git a/.github/workflows/wheels-publish.yaml b/.github/workflows/wheels-publish.yaml
index e564af35..3a8bd856 100644
--- a/.github/workflows/wheels-publish.yaml
+++ b/.github/workflows/wheels-publish.yaml
@@ -119,6 +119,11 @@ jobs:
       env:
         GH_TOKEN: ${{ github.token }}
 
+    - name: use gha-tools from PR branch
+      run: |
+        git clone https://github.com/rapidsai/gha-tools.git --branch support-sdists /tmp/gha-tools
+        echo "/tmp/gha-tools/tools" >> "$GITHUB_PATH"
+
     - name: Download wheels from artifact storage and publish to anaconda repository
       run: rapids-wheels-anaconda-github "${INPUTS_PACKAGE_NAME}" "${INPUTS_PACKAGE_TYPE}"
       env:
@@ -130,28 +135,3 @@ jobs:
         INPUTS_PACKAGE_TYPE: ${{ inputs.package-type }}
         # Used to override search key construction as needed
         WHEEL_SEARCH_KEY: ${{ inputs.publish-wheel-search-key }}
-
-    - name: Check if build is release
-      id: check_if_release
-      shell: bash
-      if: ${{ inputs.publish_to_pypi }}
-      run: |
-        if rapids-is-release-build; then
-          echo "is_release_build=true" | tee -a "${GITHUB_OUTPUT}"
-        else
-          echo "is_release_build=false" | tee -a "${GITHUB_OUTPUT}"
-        fi
-
-    - name: Publish the downloaded wheels to PyPI
-      if: ${{ inputs.publish_to_pypi && steps.check_if_release.outputs.is_release_build == 'true' }}
-      run: |
-        python3 -m pip install twine
-        python3 -m twine upload -u __token__ dist/*
-      env:
-        TWINE_PASSWORD: ${{ secrets.RAPIDSAI_PYPI_TOKEN }}
-    - name: Telemetry upload attributes
-      uses: rapidsai/shared-actions/telemetry-dispatch-stash-job-artifacts@main
-      continue-on-error: true
-      if: ${{ vars.TELEMETRY_ENABLED == 'true' }}
-      env:
-        GH_TOKEN: ${{ github.token }}

From 3306473a63ebfd42fdbd3fd6b4c7f4f33a611195 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Mon, 22 Jun 2026 13:32:39 -0500
Subject: [PATCH 2/2] fix RAPIDS_SHA

---
 .github/workflows/wheels-publish.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/wheels-publish.yaml b/.github/workflows/wheels-publish.yaml
index 3a8bd856..2124858a 100644
--- a/.github/workflows/wheels-publish.yaml
+++ b/.github/workflows/wheels-publish.yaml
@@ -125,7 +125,10 @@ jobs:
         echo "/tmp/gha-tools/tools" >> "$GITHUB_PATH"
 
     - name: Download wheels from artifact storage and publish to anaconda repository
-      run: rapids-wheels-anaconda-github "${INPUTS_PACKAGE_NAME}" "${INPUTS_PACKAGE_TYPE}"
+      run: |
+        # get RAPIDS SHA
+        export RAPIDS_SHA=$(git rev-parse HEAD)
+        rapids-wheels-anaconda-github "${INPUTS_PACKAGE_NAME}" "${INPUTS_PACKAGE_TYPE}"
       env:
         GH_TOKEN: ${{ github.token }}
         RAPIDS_CONDA_TOKEN: ${{ secrets.CONDA_RAPIDSAI_WHEELS_NIGHTLY_TOKEN }}