Merge branch 'fix-sssp-benchmark' of github.com:nv-rliu/cugraph into …

…fix-sssp-benchmark
rapidsai · Jul 17, 2024 · ceef142 · ceef142
2 parents 5cb832f + ee14d43
commit ceef142
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)