diff --git a/flax/linen/module.py b/flax/linen/module.py index 9f8c48e32b..c0c99ae6e5 100644 --- a/flax/linen/module.py +++ b/flax/linen/module.py @@ -2760,7 +2760,7 @@ def module_paths( ... }) {'': 'Foo', 'Dense_0': 'Dense', 'Dense_1': 'Dense'} - `Args: + Args: rngs: The rngs for the variable collections as passed to ``Module.init``. *args: The arguments to the forward computation. show_repeated: If ``True``, repeated calls to the same module will be