diff --git a/python/cugraph/cugraph/structure/property_graph.py b/python/cugraph/cugraph/structure/property_graph.py index 80c1bd354f..c187d66849 100644 --- a/python/cugraph/cugraph/structure/property_graph.py +++ b/python/cugraph/cugraph/structure/property_graph.py @@ -2011,17 +2011,27 @@ def edge_props_to_graph( # symmetrize function. # Symmetrize the internal representation of the edgelists - source_col, dest_col, value_col = symmetrize( - G.edgelist.edgelist_df, - "src", - "dst", - "weights", - symmetrize=not G.is_directed()) + + if edge_attr is not None: + source_col, dest_col, value_col = symmetrize( + G.edgelist.edgelist_df, + "src", + "dst", + "weights", + symmetrize=not G.is_directed()) + else: + source_col, dest_col = symmetrize( + G.edgelist.edgelist_df, + "src", + "dst", + symmetrize=not G.is_directed()) + renumbered_edge_prop_df = cudf.DataFrame() renumbered_edge_prop_df["src"] = source_col renumbered_edge_prop_df["dst"] = dest_col - renumbered_edge_prop_df["weights"] = value_col + if edge_attr: + renumbered_edge_prop_df["weights"] = value_col G.edgelist.edgelist_df = renumbered_edge_prop_df