flax
fixed docstring
#3643
Merged

Loading