Skip to content

Commit c72bbc4

Browse files
committed
Typo fix in _cached_partial method
1 parent 9048b4e commit c72bbc4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

flax/nnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1400,7 +1400,7 @@ def _cached_partial(f: tp.Callable[..., tp.Any], *cached_args):
14001400
of the cache, and these clones will contain references to the same Variable objects
14011401
which guarantees that state is propagated correctly back to the original graph nodes.
14021402
Because of the previous, the final structure of all graph nodes must be the same
1403-
after each call to the cached function, otherswise an error will be raised. Temporary
1403+
after each call to the cached function, otherwise an error will be raised. Temporary
14041404
mutations are allowed (e.g. the use of ``Module.sow``) as long as they are cleaned up before
14051405
the function returns (e.g. via ``nnx.pop``).
14061406

0 commit comments

Comments
 (0)