add columns major_vertices and minor_vertices to shuffled_df when shu…

…ffled_minor_series and shuffled_major_series are None
rapidsai · Apr 16, 2022 · 3674b7d · 3674b7d
1 parent 0b69c82
commit 3674b7d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cugraph/cugraph/structure/renumber_wrapper.pyx b/python/cugraph/cugraph/structure/renumber_wrapper.pyx
@@ -53,11 +53,11 @@ cdef renumber_helper(shuffled_vertices_t* ptr_maj_min_w, vertex_t, weights):
     # vertex_t or weight_t. Failing to do that will create am empty column of type object
     # which is not supported by '__cuda_array_interface__'
     if shuffled_major_series is None:
-        shuffled_major_series = cudf.Series(dtype=vertex_t)
+        shuffled_df['major_vertices'] = cudf.Series(dtype=vertex_t)
     else:
         shuffled_df['major_vertices']= shuffled_major_series
     if shuffled_minor_series is None:
-        shuffled_minor_series = cudf.Series(dtype=vertex_t)
+        shuffled_df['minor_vertices'] = cudf.Series(dtype=vertex_t)
     else:
         shuffled_df['minor_vertices']= shuffled_minor_series