From cfac2943953d9c34bb4cbdad19b00be4cd332d5a Mon Sep 17 00:00:00 2001 From: Ralph Liu Date: Mon, 12 Feb 2024 14:04:09 -0800 Subject: [PATCH] Use to access Series obj. Update copyright year in header --- 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