Merge pull request #3752 from chiamp:intercept

PiperOrigin-RevId: 615488628
google · Mar 13, 2024 · 50b589f · 50b589f
2 parents 9fbf69d + fca6b45
commit 50b589f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flax/linen/module.py b/flax/linen/module.py
@@ -373,7 +373,7 @@ def intercept_methods(interceptor: Interceptor):
   The following methods couldn't be intercepted:
 
   1. Methods decoratored with ``nn.nowrap``.
-  2. Dunder methods including '__eq__', '__repr__', '__init__', '__hash__', and '__post_init__'.
+  2. Dunder methods including ``__eq__``, ``__repr__``, ``__init__``, ``__hash__``, and ``__post_init__``.
   3. Module dataclass fields.
   4. Module descriptors.