From 6998732c637e312ed557846ef035762acb835bcc Mon Sep 17 00:00:00 2001 From: Jamie Townsend Date: Wed, 1 Mar 2017 18:10:14 +0000 Subject: [PATCH] Small fix for "gradient not yet implemented" error --- autograd/core.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/autograd/core.py b/autograd/core.py index 6d729229..89b2d78f 100644 --- a/autograd/core.py +++ b/autograd/core.py @@ -74,13 +74,14 @@ def __call__(self, *args, **kwargs): def vjp(self, argnum, outgrad, ans, vs, gvs, args, kwargs): try: - return self.vjps[argnum](outgrad, ans, vs, gvs, *args, **kwargs) + vjp = self.vjps[argnum] except KeyError: if self.vjps == {}: errstr = "Gradient of {0} not yet implemented." else: errstr = "Gradient of {0} w.r.t. arg number {1} not yet implemented." raise NotImplementedError(errstr.format(self.fun.__name__, argnum)) + return vjp(outgrad, ans, vs, gvs, *args, **kwargs) def defvjp(self, vjpmaker, argnum=0): vjpmaker.__name__ = "VJP_{}_of_{}".format(argnum, self.__name__)