Update to new version of halt

jax-ml · Aug 21, 2024 · 64f8b3c · 64f8b3c
1 parent 267afe1
commit 64f8b3c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/cloud-tpu-presubmit.yml b/.github/workflows/cloud-tpu-presubmit.yml
@@ -34,10 +34,9 @@ jobs:
       - name: Install JAX test requirements
         run: |
           pip install -U -r build/test-requirements.txt
-          sleep 3600
       # 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@61e7d8d6c273b102e4a6271c1e84bd0a4febc8cb
+        uses: google-ml-infra/jax-fork/actions/ci_connection@0b98fcaa920fcae8374c61b84febcfbe5c3b472f
         with:
           halt-dispatch-input: "1"
       - name: Build JAX