From b737b02f00b819a9f48a784b04a68c53942153c0 Mon Sep 17 00:00:00 2001 From: jnke2016 Date: Sun, 22 Sep 2024 19:08:42 -0700 Subject: [PATCH] deprecate python method to symmetrize the edgelist --- .../cugraph/cugraph/structure/symmetrize.py | 20 ++++++------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/python/cugraph/cugraph/structure/symmetrize.py b/python/cugraph/cugraph/structure/symmetrize.py index a8369598515..0652f05e728 100644 --- 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):