It is difficult to build a smooth workflow when both calculate_arbolate_sum and add_plus_network_attributes are needed because the former requires a ID/toID topology and the latter a comid/tocomid topology.
Would you be interested in changing add_plus_network_attributes to need the more general ID/toID set?