flax

d87f7da7 - Merge pull request #5222 from samanklesaria:doctest_check