From c777f876ac381758e8756bc74e6d521e9038a3ef Mon Sep 17 00:00:00 2001 From: jnke2016 Date: Mon, 25 Sep 2023 13:10:11 -0700 Subject: [PATCH] undo changes --- .../structure/graph_implementation/simpleDistributedGraph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py index dc31b15903e..e2424f0211e 100644 --- a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py +++ b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py @@ -181,6 +181,7 @@ def __from_edgelist( workers = _client.scheduler_info()["workers"] # Repartition to 2 partitions per GPU for memory efficient process input_ddf = input_ddf.repartition(npartitions=len(workers) * 2) + input_ddf = input_ddf.map_partitions(lambda df: df.copy()) # The dataframe will be symmetrized iff the graph is undirected # otherwise, the inital dataframe will be returned if edge_attr is not None: @@ -316,7 +317,6 @@ def __from_edgelist( is_multigraph=self.properties.multi_edge, is_symmetric=not self.properties.directed, ) - ddf = ddf.map_partitions(lambda df: df.copy()) ddf = ddf.repartition(npartitions=len(workers) * 2) ddf = persist_dask_df_equal_parts_per_worker(ddf, _client) num_edges = len(ddf)