Add a debug halt for job testing

jax-ml · Aug 21, 2024 · 0cc434b · 0cc434b
1 parent 6ad9d78
commit 0cc434b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/cloud-tpu-presubmit.yml b/.github/workflows/cloud-tpu-presubmit.yml
@@ -36,7 +36,9 @@ jobs:
           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
+        uses: google-ml-infra/jax-fork/actions/ci_connection@61e7d8d6c273b102e4a6271c1e84bd0a4febc8cb
+        with:
+          halt-dispatch-input: "1"
       - name: Build JAX
         run: |
           pip uninstall -y jaxlib