flax

1fdf7c58 - Merge pull request #1020 from jheek:fix-init-type-annotation