Stop falling back to inspect.getargspec since this is only for Python 2.

PiperOrigin-RevId: 573266994
google · Oct 18, 2023 · 00adeb1 · 00adeb1
1 parent b6872b9
commit 00adeb1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/edward2/tensorflow/program_transformations.py b/edward2/tensorflow/program_transformations.py
@@ -117,10 +117,7 @@ def _get_function_inputs(f, src_kwargs):
   if hasattr(f, "_func"):  # functions returned by tf.make_template
     f = f._func  # pylint: disable=protected-access
 
-  try:  # getargspec was deprecated in Python 3.6
-    argspec = inspect.getfullargspec(f)  # pytype: disable=module-attr
-  except AttributeError:
-    argspec = inspect.getargspec(f)
+  argspec = inspect.getfullargspec(f)
 
   fkwargs = {k: v for k, v in src_kwargs.items() if k in argspec.args}
   return fkwargs