Skip to content

Commit c3bb6ef

Browse files
committed
update column name
1 parent 2a7935e commit c3bb6ef

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/cugraph-service/server/cugraph_service_server/cugraph_handler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
1+
# Copyright (c) 2022-2025, NVIDIA CORPORATION.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.
@@ -97,9 +97,9 @@ def call_algo(sg_algo_func, G, **kwargs):
9797
kwargs_to_pass = {a: kwargs[a] for a in possible_args if a in kwargs}
9898
result_df = uniform_neighbor_sample(G, **kwargs_to_pass)
9999
# Convert DataFrame into CuPy arrays for returning to the client
100-
sources = result_df["sources"].to_cupy()
101-
destinations = result_df["destinations"].to_cupy()
102-
indices = result_df["indices"].to_cupy()
100+
sources = result_df["majors"].to_cupy()
101+
destinations = result_df["minors"].to_cupy()
102+
indices = result_df["weight"].to_cupy()
103103

104104
return UniformNeighborSampleResult(
105105
sources=sources,

0 commit comments

Comments
 (0)