From 2bdc737683f1719148f26a8ccee33f959dbb3d9f Mon Sep 17 00:00:00 2001
From: Richard Liu <39319471+richardsliu@users.noreply.github.com>
Date: Tue, 26 Nov 2024 14:37:33 -0800
Subject: [PATCH] Fix one test for nightly

Updating TF runtime environment for v4-32 in nightly test
---
 dags/solutions_team/solutionsteam_tf_nightly_supported.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
index d79bfdf10..fb67088ae 100644
--- a/dags/solutions_team/solutionsteam_tf_nightly_supported.py
+++ b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
@@ -66,7 +66,7 @@
       tpu_zone=Zone.US_CENTRAL2_B.value,
       time_out_in_min=60,
       is_pod=True,
-      runtime_version=RuntimeVersion.TPU_VM_TF_NIGHTLY_POD.value,
+      runtime_version=RuntimeVersion.V2_ALPHA_TPUV5.value,
   )
 
   tf_resnet_v5e_4 = tf_config.get_tf_resnet_config(