From 71b23ea2fb6f9dc272e9f438546d3e68dd821f73 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Tue, 17 Dec 2024 13:26:39 -0800
Subject: [PATCH] Enable PJRT compatibility in cloud TPU CI.

There's no reason to limit this to only one CI job.

PiperOrigin-RevId: 707227820
---
 .github/workflows/cloud-tpu-ci-nightly.yml | 1 -
 1 file changed, 1 deletion(-)

diff --git a/.github/workflows/cloud-tpu-ci-nightly.yml b/.github/workflows/cloud-tpu-ci-nightly.yml
index 9605365ce3aa..ec9ba4c1b9b8 100644
--- a/.github/workflows/cloud-tpu-ci-nightly.yml
+++ b/.github/workflows/cloud-tpu-ci-nightly.yml
@@ -34,7 +34,6 @@ jobs:
     name: "TPU test (jaxlib=${{ matrix.jaxlib-version }}, ${{ matrix.tpu.type }})"
     env:
       LIBTPU_OLDEST_VERSION_DATE: 20240922
-      ENABLE_PJRT_COMPATIBILITY: ${{ matrix.jaxlib-version == 'nightly+oldest_supported_libtpu' }}
       PYTHON: python${{ matrix.python-version }}
     runs-on: ${{ matrix.tpu.runner }}
     container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest"