From 655e1d5e11c43c030c0a1f9e0abffa83689e0c5d Mon Sep 17 00:00:00 2001 From: Ralph Liu Date: Mon, 12 Feb 2024 13:53:25 -0800 Subject: [PATCH] Fixes --- python/cugraph/cugraph/structure/number_map.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/cugraph/cugraph/structure/number_map.py b/python/cugraph/cugraph/structure/number_map.py index d7da20f9d84..b0118fee960 100644 --- a/python/cugraph/cugraph/structure/number_map.py +++ b/python/cugraph/cugraph/structure/number_map.py @@ -1,4 +1,4 @@ -# Copyright (c) 2020-2023, NVIDIA CORPORATION. +# Copyright (c) 2020-2024, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -497,7 +497,7 @@ def renumber_and_segment( # can't determine the edgelist input type unrenumbered_id_type = None else: - unrenumbered_id_type = df.dtypes[0] + unrenumbered_id_type = df.dtypes.iloc[0] if np.int64 in list(df.dtypes): renumber_id_type = np.int64