From 86abfd61587ded7661032b732f5bf662a296524e Mon Sep 17 00:00:00 2001 From: Ralph Liu <137829296+nv-rliu@users.noreply.github.com> Date: Tue, 16 Jul 2024 22:16:14 -0400 Subject: [PATCH] Fix Warning from `simpleDistributedGraph.py` (#4540) This PR fixes a warning seen when accessing a `dask_cudf.Series` object directly Authors: - Ralph Liu (https://github.com/nv-rliu) Approvers: - Rick Ratzel (https://github.com/rlratzel) URL: https://github.com/rapidsai/cugraph/pull/4540 --- .../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)