rm kw argument unused

pyro-ppl · Feb 8, 2024 · f86a8c6 · f86a8c6
1 parent 2d40657
commit f86a8c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/optim.py b/numpyro/optim.py
@@ -105,7 +105,7 @@ def eval_and_stable_update(
         """
         params = self.get_params(state)
         (out, aux), grads = _value_and_grad(
-            fn, x=params, has_aux=True, forward_mode_differentiation=forward_mode_differentiation
+            fn, x=params, forward_mode_differentiation=forward_mode_differentiation
         )
         out, state = lax.cond(
             jnp.isfinite(out) & jnp.isfinite(ravel_pytree(grads)[0]).all(),