Skip to content

Commit d89835a

Browse files
nitins17Google-ML-Automation
authored andcommitted
Fix matrix exclude syntax in TPU tests block
Also, skip Python 3.13 for now due to missing dependency error. PiperOrigin-RevId: 736120590
1 parent a6ab6bb commit d89835a

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

.github/workflows/wheel_tests_nightly_release.yml

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,9 @@ jobs:
6565
strategy:
6666
fail-fast: false # don't cancel all jobs on failure
6767
matrix:
68-
python: ["3.10","3.11", "3.12", "3.13"]
68+
# Skip Python 3.13 as it fails due to missing TensorFlow wheels (used for
69+
# profiler_test.py, build/collect-profile-requirements.txt) for that version (b/402590302)
70+
python: ["3.10", "3.11", "3.12"]
6971
tpu-specs: [
7072
# {type: "v3-8", cores: "4"}, # Enable when we have the v3 type available
7173
{type: "v4-8", cores: "4", runner: "linux-x86-ct4p-240-4tpu"},
@@ -76,17 +78,16 @@ jobs:
7678
- libtpu-version-type: ${{ startsWith(github.ref_name, 'release/') && 'nightly' }}
7779
- libtpu-version-type: ${{ !startsWith(github.ref_name, 'release/') && 'pypi_latest' }}
7880
# Run a single Python version for v4-8.
79-
- tpu-specs.type: "v4-8"
81+
- tpu-specs:
82+
type: "v4-8"
8083
python: "3.10"
81-
- tpu-specs.type: "v4-8"
84+
- tpu-specs:
85+
type: "v4-8"
8286
python: "3.11"
83-
- tpu-specs.type: "v4-8"
84-
python: "3.12"
8587
# Run min and max Python versions for v5e-8
86-
- tpu-specs.type: "v5e-8"
88+
- tpu-specs:
89+
type: "v5e-8"
8790
python: "3.11"
88-
- tpu-specs.type: "v5e-8"
89-
python: "3.12"
9091
name: "TPU tests (jax=${{ startsWith(github.ref_name, 'release/') && 'latest release' || 'nightly' }}, jaxlib=${{ startsWith(github.ref_name, 'release/') && 'latest release' || 'nightly' }})"
9192
with:
9293
runner: ${{ matrix.tpu-specs.runner }}

0 commit comments

Comments
 (0)