In OSS test script, use tf-keras-nightly (Keras 2) instead of Keras 3.

PiperOrigin-RevId: 576607011
tensorflow · Oct 25, 2023 · c552827 · c552827
1 parent c3586af
commit c552827
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 \