[CI] Temporarily disable TPU v6 due to runner issues

PiperOrigin-RevId: 744825924

Author: MichaelHudgins

.github/workflows/cloud-tpu-ci-nightly.yml

@@ -27,18 +27,9 @@ jobs:
         jaxlib-version: ["head", "pypi_latest", "nightly", "nightly+oldest_supported_libtpu"]
         tpu: [
           {type: "v4-8", cores: "4", runner: "linux-x86-ct4p-240-4tpu"},
-          {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"},
-          {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"}
+          {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"}
         ]
         python-version: ["3.10"]
-        # Exclude v6e-8 tests for nightly+oldest_supported_libtpu and pypi_latest for resource constraints.
-        exclude:
-          - tpu:
-              type: "v6e-8"
-            jaxlib-version: "nightly+oldest_supported_libtpu"
-          - tpu:
-              type: "v6e-8"
-            jaxlib-version: "pypi_latest"
     name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu.type }})"
     env:
       LIBTPU_OLDEST_VERSION_DATE: 20241205

.github/workflows/wheel_tests_nightly_release.yml

@@ -80,26 +80,19 @@ jobs:
           tpu-specs: [
             # {type: "v3-8", cores: "4"}, # Enable when we have the v3 type available
             {type: "v4-8", cores: "4", runner: "linux-x86-ct4p-240-4tpu"},
-            {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"},
-            {type: "v6e-8", cores: "8", runner: "linux-x86-ct6e-180-8tpu"}
+            {type: "v5e-8", cores: "8", runner: "linux-x86-ct5lp-224-8tpu"}
           ]
           libtpu-version-type: ["pypi_latest", "nightly", "oldest_supported_libtpu"]
           exclude:
             - libtpu-version-type: ${{ startsWith(github.ref_name, 'release/') && 'nightly' }}
             - libtpu-version-type: ${{ !startsWith(github.ref_name, 'release/') && 'pypi_latest' }}
-            # Run a single Python version for v4-8 and v6e-8
+            # Run a single Python version for v4-8
             - tpu-specs:
                 type: "v4-8"
               python: "3.10"
             - tpu-specs:
                 type: "v4-8"
               python: "3.11"
-            - tpu-specs:
-                type: "v6e-8"
-              python: "3.10"
-            - tpu-specs:
-                type: "v6e-8"
-              python: "3.11"
             # Run min and max Python versions for v5e-8
             - tpu-specs:
                 type: "v5e-8"