Fix TFP tests for py3.13

PiperOrigin-RevId: 817540723

Author: brianwa84

tensorflow_probability/python/internal/distribution_util.py

@@ -1281,6 +1281,7 @@ def parent_frame_arguments():
 
   # Remove the *varargs, and flatten the **kwargs. Both are
   # nested lists.
+  local_vars = dict(local_vars)
   local_vars.pop(variable_arg_name, {})
   keyword_args = local_vars.pop(keyword_arg_name, {})
 

tensorflow_probability/python/internal/unnest_test.py

@@ -163,9 +163,9 @@ def test_flat_replace(self):
     self.assertEqual(unnest.replace_outermost(
         results, return_unused=True, unique_core=2, foo=3),
                      (FakeResults(2), {'foo': 3}))
-    self.assertRaises(ValueError, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
-    self.assertRaises(ValueError, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
 
   def test_nesting_replace(self):
@@ -192,9 +192,9 @@ def build(a, b):
     self.assertEqual(unnest.replace_outermost(
         results, return_unused=True, unique_nesting=2, unique_core=3, foo=4),
                      (build(2, 3), {'foo': 4}))
-    self.assertRaises(ValueError, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
-    self.assertRaises(ValueError, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
 
   def test_atypical_nesting_replace(self):
@@ -226,9 +226,9 @@ def build(a, b):
         results, return_unused=True,
         unique_atypical_nesting=2, unique_core=3, foo=4),
                      (build(2, 3), {'foo': 4}))
-    self.assertRaises(ValueError, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
-    self.assertRaises(ValueError, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
 
   def test_deeply_nested_replace(self):
@@ -261,9 +261,9 @@ def test_deeply_nested_replace(self):
         results, return_unused=True, unique_nesting=5,
         unique_atypical_nesting=6, unique_core=7, foo=8),
                     (_build_deeply_nested(5, 6, 2, 3, 7), {'foo': 8}))
-    self.assertRaises(ValueError, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_innermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
-    self.assertRaises(ValueError, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
+    self.assertRaises(Exception, lambda: unnest.replace_outermost(   # pylint: disable=g-long-lambda
         results, unique_core=1, foo=1))
 
   def test_get_nested_obj(self):