Work around TF bug on Python 3.12.

tensorflow · Mar 11, 2024 · 7d9143a · 7d9143a
1 parent 988f023
commit 7d9143a
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=WRAPT_DISABLE_EXTENSIONS=true \
   --action_env=PATH \
   --action_env=LD_LIBRARY_PATH \
   --test_output=errors \