File tree 1 file changed +7
-4
lines changed
python/cugraph/cugraph/structure
1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -73,15 +73,17 @@ cdef coo_to_df(GraphCOOPtrType graph):
73
73
# FIXME: this function assumes columns named "src" and "dst" and can only
74
74
# be used for SG graphs due to that assumption.
75
75
contents = move(graph.get()[0 ].release())
76
- src = move_device_buffer_to_column (
76
+ src = move_device_buffer_to_series (
77
77
move(contents.src_indices),
78
78
DataType(type_id.INT32),
79
79
4 ,
80
+ None ,
80
81
)
81
- dst = move_device_buffer_to_column (
82
+ dst = move_device_buffer_to_series (
82
83
move(contents.dst_indices),
83
84
DataType(type_id.INT32),
84
85
4 ,
86
+ None ,
85
87
)
86
88
87
89
if GraphCOOPtrType is GraphCOOPtrFloat:
@@ -93,10 +95,11 @@ cdef coo_to_df(GraphCOOPtrType graph):
93
95
else :
94
96
raise TypeError (" Invalid GraphCOOPtrType" )
95
97
96
- wgt = move_device_buffer_to_column (
98
+ wgt = move_device_buffer_to_series (
97
99
move(contents.edge_data),
98
100
weight_type,
99
- itemsize
101
+ itemsize,
102
+ None ,
100
103
)
101
104
102
105
df = cudf.DataFrame()
You can’t perform that action at this time.
0 commit comments