From 17fbdff86de38f5e172555e65d63e3143ac00848 Mon Sep 17 00:00:00 2001 From: Cristian Garcia Date: Thu, 22 Feb 2024 23:45:28 +0000 Subject: [PATCH] fix Module.module_paths docstring --- flax/linen/module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flax/linen/module.py b/flax/linen/module.py index 9f8c48e32..c0c99ae6e 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