fix typo

rapidsai · Oct 1, 2024 · a9c7064 · a9c7064
1 parent d135ac1
commit a9c7064
Showing 1 changed file with 17 additions and 7 deletions.
diff --git 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