Update edward2 to Python 3.11 compatibility.

PiperOrigin-RevId: 569298509
google · Sep 28, 2023 · 3a9b7ef · 3a9b7ef
1 parent d23eddd
commit 3a9b7ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/edward2/numpy/program_transformations.py b/edward2/numpy/program_transformations.py
@@ -149,9 +149,9 @@ def _get_function_inputs(f, *args, **kwargs):
     New original args, args of f, kwargs of f.
   """
   if hasattr(f, "_func"):  # functions returned by tf.make_template
-    argspec = inspect.getargspec(f._func)  # pylint: disable=protected-access
+    argspec = inspect.getfullargspec(f._func)  # pylint: disable=protected-access
   else:
-    argspec = inspect.getargspec(f)
+    argspec = inspect.getfullargspec(f)
 
   fkwargs = {}
   for k, v in kwargs.items():