From 6bd59d1e5e494f63f1cf729c8bbd30007a91eef9 Mon Sep 17 00:00:00 2001 From: jnke2016 Date: Tue, 12 Mar 2024 17:28:13 -0700 Subject: [PATCH] fix typo --- python/cugraph/cugraph/sampling/node2vec.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cugraph/cugraph/sampling/node2vec.py b/python/cugraph/cugraph/sampling/node2vec.py index 7bdac2a2abe..71fc2969f86 100644 --- a/python/cugraph/cugraph/sampling/node2vec.py +++ b/python/cugraph/cugraph/sampling/node2vec.py @@ -24,11 +24,11 @@ # FIXME: Move this function to the utility module so that it can be # shared by other algos def ensure_valid_dtype(input_graph, start_vertices): - vertex_dtype = input_graph.edgelist.edgelist_df.iloc[0] + vertex_dtype = input_graph.edgelist.edgelist_df.dtypes.iloc[0] if isinstance(start_vertices, cudf.Series): start_vertices_dtype = start_vertices.dtype else: - start_vertices_dtype = start_vertices.iloc[0] + start_vertices_dtype = start_vertices.dtypes.iloc[0] if start_vertices_dtype != vertex_dtype: warning_msg = (