Skip to content

Commit

Permalink
Flatten cupy array before feeding to cudf.Series (#5651)
Browse files Browse the repository at this point in the history
Previously it seems that cudf was silently flattening 2D arrays when passing them to the cudf.Series constructor, but that is no longer supported here so the test code needs to be updated.

Authors:
  - Vyas Ramasubramani (https://github.com/vyasr)

Approvers:
  - Simon Adorf (https://github.com/csadorf)

URL: #5651
  • Loading branch information
vyasr authored Nov 9, 2023
1 parent 9fed69e commit 6d51185
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/cuml/tests/test_input_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -396,7 +396,7 @@ def get_input(
result = cudf.DataFrame(rand_mat, index=index)

if type == "cudf-series":
result = cudf.Series(rand_mat, index=index)
result = cudf.Series(rand_mat.reshape(nrows), index=index)

if type == "pandas":
result = pdDF(cp.asnumpy(rand_mat), index=index)
Expand Down

0 comments on commit 6d51185

Please sign in to comment.