diff --git a/flax/linen/kw_only_dataclasses.py b/flax/linen/kw_only_dataclasses.py index 84d2ec0032..03499a9f37 100644 --- 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.