From b10b23e9f02b451ddb0b8f5ec5c34f7e6004f813 Mon Sep 17 00:00:00 2001 From: mhucka Date: Thu, 22 Jan 2026 22:40:08 +0000 Subject: [PATCH 1/2] Use larger runners --- .github/workflows/ci-nightly-build-test.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci-nightly-build-test.yaml b/.github/workflows/ci-nightly-build-test.yaml index 7ef135904..a4746385e 100644 --- a/.github/workflows/ci-nightly-build-test.yaml +++ b/.github/workflows/ci-nightly-build-test.yaml @@ -93,7 +93,7 @@ jobs: if: needs.Decision.outputs.run == 'true' name: Build and test needs: Decision - runs-on: ubuntu-22.04 + runs-on: linux-x86-n2-32 steps: - name: Check out a copy of the TFQ git repository uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4 From 4cdc6a0be635ed0f0dd4da8b8ff16588ed2dbdb9 Mon Sep 17 00:00:00 2001 From: mhucka Date: Mon, 26 Jan 2026 20:47:37 +0000 Subject: [PATCH 2/2] Add a pinned version of the Docker image used for the runner The ml-build Docker image selected by the tag used here is Ubuntu 22.04 That image is created by ci/official/containers/ml_build/Dockerfile in the TensorFlow GitHub repository (specifically, the version of the file as it existed at git commit 1b4ef03f370aa7b266722e1055800275a2db11d0). --- .github/workflows/ci-nightly-build-test.yaml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/ci-nightly-build-test.yaml b/.github/workflows/ci-nightly-build-test.yaml index a4746385e..6a711a24f 100644 --- a/.github/workflows/ci-nightly-build-test.yaml +++ b/.github/workflows/ci-nightly-build-test.yaml @@ -93,6 +93,8 @@ jobs: if: needs.Decision.outputs.run == 'true' name: Build and test needs: Decision + container: + image: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:infrastructure-public-image-82c7ac59ba4b runs-on: linux-x86-n2-32 steps: - name: Check out a copy of the TFQ git repository