Make static type checkers aware the fact that all derived classes of …

…`FiddleBaseParameterizable` have a dataclass-like semantics (ie having fields). PiperOrigin-RevId: 573301464
google · Oct 13, 2023 · 7805111 · 7805111
1 parent fac0cef
commit 7805111
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flax/linen/kw_only_dataclasses.py b/flax/linen/kw_only_dataclasses.py
@@ -99,6 +99,7 @@ def field(*, metadata=None, kw_only=dataclasses.MISSING, **kwargs):
   return dataclasses.field(metadata=metadata, **kwargs)
 
 
+# LINT.IfChange
 @dataclass_transform(field_specifiers=(field,))  # type: ignore[literal-required]
 def dataclass(cls=None, extra_fields=None, **kwargs):
   """Wrapper for dataclasses.dataclass that adds support for kw_only fields.