flax
Fix doctests in transforms for graph_updates=False default
#5441
Merged

Loading