[BUG] Visualiser does not expand all boxes #373
Description
Describe the bug
In certain circumstances the visualiser will not expand all of the expandable nodes in the graph.
To Reproduce
See attached checkpoint dir. Open this in the visualiser. Try expanding the boxes in order: loop, loop iteration, first eval, second eval, ... no subsequent boxes will expand.
Expected behavior
It should be possible to expand all of the boxes.