Include connection halt into tpu presubmit

jax-ml · Aug 20, 2024 · 683526b · 683526b
1 parent 6d6a19b
commit 683526b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/cloud-tpu-presubmit.yml b/.github/workflows/cloud-tpu-presubmit.yml
@@ -35,6 +35,8 @@ jobs:
         run: |
           pip install -U -r build/test-requirements.txt
       # TODO: build jax should be done on a step prior or we should just bazel test
+      - name: Wait For Connection
+        uses: google-ml-infra/jax-fork/actions/ci_connection@28deea80eeffa1e98ba84ebb29200b7b00c1b61f
       - name: Build JAX
         run: |
           pip uninstall -y jaxlib