From a4af89b3dd20ea43ca81654869195d8b547981a8 Mon Sep 17 00:00:00 2001
From: Graham Markall <gmarkall@nvidia.com>
Date: Tue, 23 Jul 2024 17:52:43 +0100
Subject: [PATCH] Upload wheel to artifact storage

---
 .github/workflows/wheels-build.yaml | 6 +++++-
 ci/build_wheel.sh                   | 4 ++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/wheels-build.yaml b/.github/workflows/wheels-build.yaml
index f860564..7efcd3f 100644
--- a/.github/workflows/wheels-build.yaml
+++ b/.github/workflows/wheels-build.yaml
@@ -158,5 +158,9 @@ jobs:
         # Use a shell that loads the rc file so that we get the compiler settings
         shell: bash -leo pipefail {0}
       - name: Upload additional artifacts
+        id: upload-wheel
         if: "!cancelled()"
-        run: rapids-upload-artifacts-dir cuda${RAPIDS_CUDA_VERSION%%.*}_$(arch)_py${RAPIDS_PY_VERSION//.}
+        uses: actions/upload-artifact@v4
+        with:
+          name ${{ env.package_name }}
+          path: ${{ env.package_path }}/${{ env.package_name }}
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
index 949fb30..140974a 100755
--- a/ci/build_wheel.sh
+++ b/ci/build_wheel.sh
@@ -7,5 +7,5 @@ rapids-logger "Build wheel"
 mkdir -p ./dist
 python -m pip wheel . --wheel-dir=./dist -vvv --disable-pip-version-check --no-deps
 
-rapids-logger "Upload Wheel"
-RAPIDS_PY_WHEEL_NAME="numba_cuda" rapids-upload-wheels-to-s3 ./dist
+echo "package_name=$(basename numba_cuda-*.whl)" >> $GITHUB_ENV
+echo "package_path=`pwd`" >> $GITHUB_ENV