Skip to content

Commit

Permalink
Fix FutureWarnings in Graph Creation (#4167)
Browse files Browse the repository at this point in the history
Addresses #4163 

This PR uses `iloc` when referencing the index in a Series object, which removes the `FutureWarning` that was seen at graph creation.

Authors:
  - Ralph Liu (https://github.com/nv-rliu)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: #4167
  • Loading branch information
nv-rliu authored Feb 13, 2024
1 parent 9a6ad49 commit fe69e7f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions python/cugraph/cugraph/structure/number_map.py
Original file line number Diff line number Diff line change
@@ -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.
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit fe69e7f

Please sign in to comment.