Same as above. Consider returning both `flow_value` and the node partition.