Move publish step out of wheels-build.yaml

Follow-up to NVIDIA#142 (comment)

Author: KyleFromNVIDIA

.github/workflows/pr.yaml

@@ -77,6 +77,25 @@ jobs:
       build_type: pull-request
       script: "ci/build_wheel.sh"
       matrix_filter: ${{ needs.compute-matrix.outputs.BUILD_MATRIX }}
+  # TODO: Remove this job before merging
+  fake-publish-wheels:
+    needs: build-wheels
+    runs-on: ubuntu-latest
+    steps:
+      - name: Download sdist
+        if: "!cancelled()"
+        uses: actions/download-artifact@v4
+        with:
+          name: sdist
+          path: dist
+      - name: Download wheel
+        if: "!cancelled()"
+        uses: actions/download-artifact@v4
+        with:
+          name: wheel
+          path: dist
+      - name: List wheels
+        run: ls dist/
   test-wheels:
     needs:
       - build-wheels

.github/workflows/publish.yaml

@@ -41,7 +41,26 @@ jobs:
       build_type: release
       script: "ci/build_wheel.sh"
       matrix_filter: ${{ needs.compute-matrix.outputs.BUILD_MATRIX }}
-      upload_to_pypi: true
+  publish-wheels:
+    needs: build-wheels
+    runs-on: ubuntu-latest
+    steps:
+      - name: Download sdist
+        if: "!cancelled()"
+        uses: actions/download-artifact@v4
+        with:
+          name: sdist
+          path: dist
+      - name: Download wheel
+        if: "!cancelled()"
+        uses: actions/download-artifact@v4
+        with:
+          name: wheel
+          path: dist
+      - name: Publish wheel
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          password: ${{ secrets.RAPIDSAI_PYPI_TOKEN }}
   build-docs:
     needs:
       - build-conda

.github/workflows/wheels-build.yaml

@@ -45,10 +45,6 @@ on:
         type: string
         default: ''
 
-      upload_to_pypi:
-        type: boolean
-        required: false
-
 defaults:
   run:
     shell: bash
@@ -173,25 +169,3 @@ jobs:
         with:
           name: wheel
           path: ${{ env.wheel_path }}
-
-  publish:
-    needs: [build]
-    runs-on: ubuntu-latest
-    if: inputs.upload_to_pypi
-    steps:
-      - name: Download sdist
-        if: "!cancelled()"
-        uses: actions/download-artifact@v4
-        with:
-          name: sdist
-          path: dist
-      - name: Download wheel
-        if: "!cancelled()"
-        uses: actions/download-artifact@v4
-        with:
-          name: wheel
-          path: dist
-      - name: Publish wheel
-        uses: pypa/gh-action-pypi-publish@release/v1
-        with:
-          password: ${{ secrets.RAPIDSAI_PYPI_TOKEN }}