Fixes so that tests pass under Python 3.12.

NOTE: WRAPT_DISABLE_EXTENSIONS=true is to work around a bug in TensorFlow + wrapt that causes code like the following to fail:
```
import tensorflow as tf

class C(tf.Module): pass

module = C()
module._some_tuple = (tf.Variable(0.),)
module.trainable_variables
# fails with: "TypeError: this __dict__ descriptor does not support '_TupleWrapper' objects"
```
(This bug causes many TFP tests to fail under Python 3.12 .)
PiperOrigin-RevId: 614658578

Author: jburnim

tensorflow_probability/python/bijectors/bijector_test.py

@@ -56,9 +56,9 @@ class BaseBijectorTest(test_util.TestCase):
   """Tests properties of the Bijector base-class."""
 
   def testIsAbstract(self):
-    with self.assertRaisesRegexp(TypeError,
-                                 ('Can\'t instantiate abstract class Bijector '
-                                  'with abstract methods? __init__')):
+    with self.assertRaisesRegex(TypeError,
+                                ('Can\'t instantiate abstract class Bijector '
+                                 'with.* abstract methods? \'?__init__')):
       bijector_lib.Bijector()  # pylint: disable=abstract-class-instantiated
 
   def testDefaults(self):

tensorflow_probability/python/distributions/joint_distribution_coroutine_test.py

@@ -1307,7 +1307,7 @@ def dist():
     warnings.simplefilter('always')
     with warnings.catch_warnings(record=True) as w:
       d.sample(seed=test_util.test_seed())
-    self.assertRegexpMatches(
+    self.assertRegex(
         str(w[0].message),
         r'Falling back to stateful sampling for distribution #1.*'
         r'of type.*StatefulNormal.*component name "loc" and `dist.name` '

tensorflow_probability/python/mcmc/BUILD

@@ -630,6 +630,7 @@ py_test(
     name = "eight_schools_hmc_eager_test",
     size = "medium",  # Might run > 1 minute.
     srcs = ["eight_schools_hmc_eager_test.py"],
+    tags = ["no-oss-ci"],
     deps = [
         ":eight_schools_hmc",
         # tensorflow dep,
@@ -642,6 +643,7 @@ py_test(
     name = "eight_schools_hmc_graph_test",
     size = "medium",  # Might run > 1 minute.
     srcs = ["eight_schools_hmc_graph_test.py"],
+    tags = ["no-oss-ci"],
     deps = [
         ":eight_schools_hmc",
         # tensorflow dep,

tensorflow_probability/python/mcmc/metropolis_hastings_test.py

@@ -385,10 +385,10 @@ def testWarnings(self):
       _, _ = mh.one_step(current_state, init_kernel_results,
                          seed=test_util.test_seed())
     w = sorted(w, key=lambda w: str(w.message))
-    self.assertRegexpMatches(
+    self.assertRegex(
         str(w[0].message),
         r'`TransitionKernel` is already calibrated')
-    self.assertRegexpMatches(
+    self.assertRegex(
         str(w[1].message),
         r'`TransitionKernel` does not have a `log_acceptance_correction`')
 

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 \