From 2cbbcd9d5e1e5d6659309fe36833796794ca8d5d Mon Sep 17 00:00:00 2001
From: Jake Awe <jawe@nvidia.com>
Date: Mon, 24 Jun 2024 13:10:08 -0500
Subject: [PATCH] add publish to pr workflow

---
 .github/workflows/pr.yaml | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml
index 5020f00c3..d2b42b510 100644
--- a/.github/workflows/pr.yaml
+++ b/.github/workflows/pr.yaml
@@ -66,3 +66,15 @@ jobs:
     with:
       build_type: pull-request
       script: ci/test_wheel.sh
+  wheel-publish:
+    permissions:
+      id-token: write
+    needs: wheel-build
+    secrets: inherit
+    uses: rapidsai/shared-workflows/.github/workflows/wheels-publish.yaml@test_pypi_trusted_publish
+    with:
+      build_type: ${{ inputs.build_type || 'branch' }}
+      branch: ${{ inputs.branch }}
+      sha: ${{ inputs.sha }}
+      date: ${{ inputs.date }}
+      package-name: cucim