flax
bc31315c - Don't wrap user-defined __post_init__ function.

Commit
4 years ago
Don't wrap user-defined __post_init__ function. To make setup() sound under laziness we removed the ability to "normalize" dataclass attributes from setup(). For those requiring this feature, allow the use of a custom __post_init__ (which of course must call super().__post_init__) by not accidentally breaking it with our automatic method wrapping. PiperOrigin-RevId: 367076810
Author
Committer
Parents
Loading