flax

b8befd49 - Merge pull request #599 from avital:fix-deprecation-warning