Merge branch 'branch-24.08' of https://github.com/rapidsai/cugraph in…

…to enh_graph_creation
rapidsai · Jul 17, 2024 · 9bdbfdb · 9bdbfdb
2 parents 954218f + 86abfd6
commit 9bdbfdb
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)