flax

f7545411 - Merge pull request #2372 from cgarciae:test-docstrings