deprecate python method to symmetrize the edgelist

rapidsai · Sep 23, 2024 · b737b02 · b737b02
1 parent f09fd41
commit b737b02
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/python/cugraph/cugraph/structure/symmetrize.py b/python/cugraph/cugraph/structure/symmetrize.py
@@ -184,6 +184,12 @@ def symmetrize_ddf(
         result = result.reset_index(drop=True).repartition(npartitions=len(workers) * 2)
         return result
     else:
+        warnings.warn(
+            "Multi is deprecated and the removal of multi edges will no longer be "
+            "supported from 'symmetrize'. Multi edges will be removed upon creation "
+            "of graph instance.",
+            FutureWarning,
+        )
         vertex_col_name = src_name + dst_name
         result = _memory_efficient_drop_duplicates(
             result, vertex_col_name, len(workers)
@@ -251,25 +257,12 @@ def symmetrize(
     >>> df['values'] = cudf.Series(M['2'])
     >>> src, dst, val = symmetrize(df, 'sources', 'destinations', 'values', multi=True)
     """
-    print("multi = \n", multi)
-    return (
-                input_df[source_col_name],
-                input_df[dest_col_name],
-                input_df[value_col_name],
-            )
-
 
-
-
-    """
     warnings.warn(
         "This method is deprecated and will no longer be supported. The symmetrization "
         "of the edges are only supported by setting the 'symmetrize' flag to 'True'",
         FutureWarning,
     )
-    """
-
-    """
 
     # FIXME: Redundant check that should be done at the graph creation
     if "edge_id" in input_df.columns and symmetrize:
@@ -313,7 +306,6 @@ def symmetrize(
             )
 
     return output_df[source_col_name], output_df[dest_col_name]
-    """
 
 
 def _add_reverse_edges(df, src_name, dst_name, weight_name):