This change is needed to allow using @flax.struct.dataclass on a class that inherits itself from another dataclass created with @flax.struct.dataclass.
This change is needed to allow using @flax.struct.dataclass on a class that inherits itself from another dataclass created with @flax.struct.dataclass.

PiperOrigin-RevId: 361753164

google/flax,google/flax