Decorating the _build function with memoize is breaking the connected_subgraph code. This CL fixes this issue. Note sure if this use-case is valid in the first place but let's have this discussion later!
Decorating the _build function with memoize is breaking the connected_subgraph code. This CL fixes this issue. Note sure if this use-case is valid in the first place but let's have this discussion later!

PiperOrigin-RevId: 171134926

rakshit-agrawal/sonnet,deepmind/sonnet,deepmind/sonnet,rakshit-agrawal/sonnet