From 5d759df0d6b2fadb6cbce3749ae5c2e11a3a7681 Mon Sep 17 00:00:00 2001 From: Ralph Liu Date: Tue, 16 Jul 2024 13:58:33 -0700 Subject: [PATCH] Fix dask warning --- .../structure/graph_implementation/simpleDistributedGraph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py index e50462c001e..7f3f7e83e59 100644 --- a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py +++ b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py @@ -891,9 +891,9 @@ def _call_plc_two_hop_neighbors(sID, mg_graph_x, start_vertices): if start_vertices is not None: if self.renumbered: start_vertices = self.renumber_map.to_internal_vertex_id(start_vertices) - start_vertices_type = self.edgelist.edgelist_df.dtypes[0] + start_vertices_type = self.edgelist.edgelist_df.dtypes.iloc[0] else: - start_vertices_type = self.input_df.dtypes[0] + start_vertices_type = self.input_df.dtypes.iloc[0] start_vertices = start_vertices.astype(start_vertices_type)