flax
Fix update doctest for graph_updates=False default
#5440
Merged

Loading