Try splitting wheel builds by python version.

PiperOrigin-RevId: 721980105
jax-ml · Feb 1, 2025 · 5b0e41d · 5b0e41d
1 parent 00d98cd
commit 5b0e41d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -21,11 +21,16 @@ concurrency:
 
 jobs:
   build_wheels:
-    name: Build wheels on ${{ matrix.os }}
+    name: Build wheels on ${{ matrix.os }} for Python ${{ matrix.cibw_build }}
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
         os: [ubuntu-20.04, macos-14, windows-2019]
+        cibw_build: [cp39, cp310, cp311, cp312, cp313, cp313t]
+        # Skip non-existing configurations.
+        exclude:
+          - os: windows-2019
+            cibw_build: cp313t
 
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683  # v4.2.2
@@ -51,7 +56,7 @@ jobs:
         env:
           CIBW_ARCHS_LINUX: auto aarch64
           CIBW_ARCHS_MACOS: universal2
-          CIBW_BUILD: cp39-* cp310-* cp311-* cp312-* cp313-* cp313t-*
+          CIBW_BUILD: ${{ matrix.cibw_build }}-*
           CIBW_FREE_THREADED_SUPPORT: True
           CIBW_PRERELEASE_PYTHONS: True
           CIBW_SKIP: "*musllinux* *i686* *win32* *t-win*"