flatten# class brainstate.graph.flatten(node, /, ref_index=None, treefy_state=True)# Flatten a graph node into a (graph_def, state_mapping) pair. Return type: tuple[GraphDef[Any], NestedDict]