From 683526b1e164f4f23acaf4e2b7c4c0cf8e4b28b3 Mon Sep 17 00:00:00 2001 From: Michael Hudgins Date: Tue, 20 Aug 2024 14:13:22 +0000 Subject: [PATCH] Include connection halt into tpu presubmit --- .github/workflows/cloud-tpu-presubmit.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/cloud-tpu-presubmit.yml b/.github/workflows/cloud-tpu-presubmit.yml index 406de87aee06..ac595afdabbb 100644 --- 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