Merge pull request #578 from musicinmybrain/getargspec

Replace inspect.getargspec on Python 3
HIPS · Jun 15, 2022 · cb5f840 · cb5f840
2 parents fc1bfa4 + 116f738
commit cb5f840
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/autograd/differential_operators.py b/autograd/differential_operators.py
@@ -2,7 +2,10 @@
 from __future__ import absolute_import
 from functools import partial
 from collections import OrderedDict
-from inspect import getargspec
+try:
+    from inspect import getfullargspec as _getargspec  # Python 3
+except ImportError:
+    from inspect import getargspec as _getargspec  # Python 2
 import warnings
 
 from .wrap_util import unary_to_nary
@@ -69,7 +72,7 @@ def holomorphic_grad(fun, x):
 def grad_named(fun, argname):
     '''Takes gradients with respect to a named argument.
        Doesn't work on *args or **kwargs.'''
-    arg_index = getargspec(fun).args.index(argname)
+    arg_index = _getargspec(fun).args.index(argname)
     return grad(fun, arg_index)
 
 @unary_to_nary