From 7d9143a216c98e5a9f6dd80b0cef3daba8b6da4e Mon Sep 17 00:00:00 2001
From: Jacob Burnim <jburnim@google.com>
Date: Mon, 11 Mar 2024 04:06:26 +0000
Subject: [PATCH] Work around TF bug on Python 3.12.

---
 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..15c1cee650 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=WRAPT_DISABLE_EXTENSIONS=true \
   --action_env=PATH \
   --action_env=LD_LIBRARY_PATH \
   --test_output=errors \