flax

6172e69b - Merge pull request #5382 from samanklesaria:deprecation_func