flax
Fix `_ParentType` annotation
#4120
Merged

Loading