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