Skip to content

Commit e879874

Browse files
committed
Ensure pylibcudf Columns are wrapped in Series if assigned
1 parent 008f9f8 commit e879874

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

python/cugraph/cugraph/structure/graph_primtypes.pyx

+7-4
Original file line numberDiff line numberDiff line change
@@ -73,15 +73,17 @@ cdef coo_to_df(GraphCOOPtrType graph):
7373
# FIXME: this function assumes columns named "src" and "dst" and can only
7474
# be used for SG graphs due to that assumption.
7575
contents = move(graph.get()[0].release())
76-
src = move_device_buffer_to_column(
76+
src = move_device_buffer_to_series(
7777
move(contents.src_indices),
7878
DataType(type_id.INT32),
7979
4,
80+
None,
8081
)
81-
dst = move_device_buffer_to_column(
82+
dst = move_device_buffer_to_series(
8283
move(contents.dst_indices),
8384
DataType(type_id.INT32),
8485
4,
86+
None,
8587
)
8688

8789
if GraphCOOPtrType is GraphCOOPtrFloat:
@@ -93,10 +95,11 @@ cdef coo_to_df(GraphCOOPtrType graph):
9395
else:
9496
raise TypeError("Invalid GraphCOOPtrType")
9597

96-
wgt = move_device_buffer_to_column(
98+
wgt = move_device_buffer_to_series(
9799
move(contents.edge_data),
98100
weight_type,
99-
itemsize
101+
itemsize,
102+
None,
100103
)
101104

102105
df = cudf.DataFrame()

0 commit comments

Comments
 (0)