From 607b948dd2f868052e1e8f9eec9d8bfd77b9398e Mon Sep 17 00:00:00 2001 From: Nikhil Bansal Date: Mon, 22 Jun 2026 11:39:25 -0700 Subject: [PATCH] OSS multiprocess unit tests PiperOrigin-RevId: 936167047 --- .github/workflows/build.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index ae3a8bc52..63f0e0fbd 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -266,7 +266,7 @@ jobs: multiprocess-unit-tests: name: "multiprocess-unit-tests (Python ${{ matrix.python-version }}, jax=${{ matrix.jax-version }})" - runs-on: linux-x86-ct5lp-224-8tpu + runs-on: linux-x86-ct6e-180-8tpu container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-2d2a7b1e6e2e defaults: run: @@ -274,7 +274,7 @@ jobs: strategy: matrix: python-version: ["3.11"] - jax-version: ["0.10.1"] + jax-version: ["newest"] steps: - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2 - name: Set up Python ${{ matrix.python-version }}