flax

e320e11c - Merge pull request #2446 from jheek:remove-pytype-generic-workaround