From c55282703c8f1dfaf1509ef8ce70127dedb55bcc Mon Sep 17 00:00:00 2001
From: jburnim <jburnim@google.com>
Date: Wed, 25 Oct 2023 12:17:19 -0700
Subject: [PATCH] In OSS test script, use tf-keras-nightly (Keras 2) instead of
 Keras 3.

PiperOrigin-RevId: 576607011
---
 testing/run_tfp_test.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/testing/run_tfp_test.sh b/testing/run_tfp_test.sh
index ce96c2840c..d7ec5a9ff3 100755
--- a/testing/run_tfp_test.sh
+++ b/testing/run_tfp_test.sh
@@ -66,6 +66,7 @@ bazel test \
   --test_timeout 300,450,1200,3600 \
   --test_tag_filters="-gpu,-requires-gpu-nvidia,-notap,-no-oss-ci,-tf2-broken,-tf2-kokoro-broken" \
   --test_env=TFP_HYPOTHESIS_MAX_EXAMPLES=2 \
+  --test_env=TF_USE_LEGACY_KERAS=1 \
   --action_env=PATH \
   --action_env=LD_LIBRARY_PATH \
   --test_output=errors \