additional linting

google-deepmind · Feb 15, 2024 · 8a478c2 · 8a478c2
1 parent 24cd16c
commit 8a478c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optax/_src/wrappers.py b/optax/_src/wrappers.py
@@ -571,7 +571,8 @@ class MaybeUpdateState(NamedTuple):
   """Maintains inner transform state and adds a step counter.
   
   Attributes:
-    inner_state (:class:`optax.OptState`): The state of the inner `GradientTransformation`.
+    inner_state (:class:`optax.OptState`): The state of the inner
+      `GradientTransformation`.
     step (``Union[jax.Array, int]``): The current step counter.
   """
   inner_state: base.OptState