add SG support for dropping multi-edges through the CAPI

rapidsai · Jan 16, 2024 · 883644c · 883644c
1 parent aa66a32
commit 883644c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py
@@ -298,7 +298,8 @@ def __from_edgelist(
             self._replicate_edgelist()
 
         self._make_plc_graph(
-            value_col=value_col, store_transposed=store_transposed, renumber=renumber
+            value_col=value_col, store_transposed=store_transposed,
+            renumber=renumber, drop_multi_edges=self.properties.multi_edge,
         )
 
     def to_pandas_edgelist(
@@ -1084,6 +1085,7 @@ def _make_plc_graph(
         value_col: Dict[str, cudf.DataFrame] = None,
         store_transposed: bool = False,
         renumber: bool = True,
+        drop_multi_edges: bool = False,
     ):
         """
         Parameters
@@ -1163,6 +1165,7 @@ def _make_plc_graph(
             renumber=renumber,
             do_expensive_check=True,
             input_array_format=input_array_format,
+            drop_multi_edges=drop_multi_edges,
         )
 
     def to_directed(self, DiG, store_transposed=False):