From 0a479e1b5c2ac66c9f2be3d92ae6b6688c037eec Mon Sep 17 00:00:00 2001 From: TAKEKAWA Takashi Date: Sun, 13 Feb 2022 23:54:59 +0000 Subject: [PATCH] Avoid custom_gradient warnings. --- tensorflow_probability/python/internal/custom_gradient.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow_probability/python/internal/custom_gradient.py b/tensorflow_probability/python/internal/custom_gradient.py index 5cbf4896af..4931b41b5f 100644 --- a/tensorflow_probability/python/internal/custom_gradient.py +++ b/tensorflow_probability/python/internal/custom_gradient.py @@ -88,7 +88,7 @@ def f_wrapped(*args, **kwargs): args = args[1:] val, aux = vjp_fwd(*reconstruct_args, **kwargs) - def vjp_bwd_wrapped(*g, **kwargs): + def vjp_bwd_wrapped(*g): # We don't want to use an explicit `variables` arg, because TF will # complain if the wrapped function doesn't actually have variables # in it. TF will only specify this arg if there are variables.