Allow dry run for PyPI release workflow

Author: zcbenz

.github/workflows/release.yml

@@ -6,29 +6,28 @@ on:
       - 'v*'
   workflow_dispatch:
     inputs:
+      dry_run:
+        description: 'Dry run (use "false" to publish to PyPI)'
+        required: true
+        default: 'false'
       dev_release:
-        description: "Do a dev release or regular release"
+        description: 'Dev release (use "false" to do a regular release)'
         required: true
-        default: "false"
+        default: 'false'
 
 permissions:
   contents: read
 
 jobs:
-  setup:
-    runs-on: ubuntu-latest
-    steps:
-      - name: Set publishing variables
-        run: echo "Publishing setup complete"
-
   build_documentation:
     if: github.repository == 'ml-explore/mlx'
     runs-on: ubuntu-22.04
     steps:
       - uses: actions/checkout@v6
       - uses: ./.github/actions/build-docs
-    
+
   deploy_documentation:
+    if: github.event.inputs.dry_run == 'false'
     needs: build_documentation
     permissions:
       pages: write
@@ -77,7 +76,7 @@ jobs:
           name: mlx-cpu-${{ matrix.arch }}
           path: wheelhouse/mlx_cpu-*.whl
           if-no-files-found: error
-  
+
   build_mac_release:
     if: github.repository == 'ml-explore/mlx'
     strategy:
@@ -87,7 +86,6 @@ jobs:
     env:
       PYPI_RELEASE: 1
       DEV_RELEASE: ${{ github.event.inputs.dev_release == 'true' && 1 || 0 }}
-
     steps:
       - uses: actions/checkout@v6
       - uses: ./.github/actions/setup-macos
@@ -162,7 +160,7 @@ jobs:
   pypi-publish:
     name: Upload release to PyPI
     runs-on: ubuntu-latest
-    needs: [setup, build_linux_release, build_mac_release]
+    needs: [build_linux_release, build_mac_release]
     permissions:
       id-token: write
     environment:
@@ -180,16 +178,17 @@ jobs:
           merge-multiple: true
           path: dist
       - name: Display structure of downloaded files
-        run: ls -R dist
+        run: ls -lR dist
       - name: Publish package distributions to PyPI
+        if: github.event.inputs.dry_run == 'false'
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           repository-url: https://upload.pypi.org/legacy/
-  
+
   pypi-publish-cuda:
     name: Upload CUDA release to PyPI
     runs-on: ubuntu-latest
-    needs: [setup, build_cuda_release]
+    needs: [build_cuda_release]
     permissions:
       id-token: write
     environment:
@@ -201,16 +200,17 @@ jobs:
           name: mlx-cuda-*
           path: dist
       - name: Display structure of downloaded files
-        run: ls -R dist
+        run: ls -lR dist
       - name: Publish package distributions to PyPI
+        if: github.event.inputs.dry_run == 'false'
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           repository-url: https://upload.pypi.org/legacy/
 
   pypi-publish-cpu:
     name: Upload CPU release to PyPI
     runs-on: ubuntu-latest
-    needs: [setup, build_linux_release]
+    needs: [build_linux_release]
     permissions:
       id-token: write
     environment:
@@ -223,16 +223,17 @@ jobs:
           merge-multiple: true
           path: dist
       - name: Display structure of downloaded files
-        run: ls -R dist
+        run: ls -lR dist
       - name: Publish package distributions to PyPI
+        if: github.event.inputs.dry_run == 'false'
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           repository-url: https://upload.pypi.org/legacy/
 
   pypi-publish-metal:
     name: Upload Metal release to PyPI
     runs-on: ubuntu-latest
-    needs: [setup, build_mac_release]
+    needs: [build_mac_release]
     permissions:
       id-token: write
     environment:
@@ -244,8 +245,9 @@ jobs:
           name: mlx-metal
           path: dist
       - name: Display structure of downloaded files
-        run: ls -R dist
+        run: ls -lR dist
       - name: Publish package distributions to PyPI
+        if: github.event.inputs.dry_run == 'false'
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           repository-url: https://upload.pypi.org/legacy/