Merge pull request #1079 from google-deepmind:fabianp-patch-16

PiperOrigin-RevId: 681084456
google-deepmind · Oct 1, 2024 · 474b4fd · 474b4fd
2 parents 5637db1 + 6838c47
commit 474b4fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optax/_src/base.py b/optax/_src/base.py
@@ -161,7 +161,7 @@ class GradientTransformation(NamedTuple):
 
   Note that an extended API is provided for users wishing to build optimizers
   that take additional arguments during the update step. For more details,
-  see ``GradientTransoformationExtraArgs``.
+  see :func:`optax.GradientTransoformationExtraArgs`.
 
   Since gradient transformations do not contain any internal state, all stateful
   optimizer properties (such as the current step count when using optimizer