Skip to content

Commit

Permalink
Adjust assumption regarding valid cudf.Series dimensional input. (#5654)
Browse files Browse the repository at this point in the history
cudf.Series does not accept any multi-dimensional input anymore.

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

Approvers:
  - Vyas Ramasubramani (https://github.com/vyasr)
  - William Hicks (https://github.com/wphicks)

URL: #5654
  • Loading branch information
csadorf authored Nov 14, 2023
1 parent a5b839f commit f2e9459
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions python/cuml/testing/strategies.py
Original file line number Diff line number Diff line change
Expand Up @@ -187,9 +187,7 @@ def create_cuml_array_input(input_type, dtype, shape, order):

input_type = "cupy" if input_type is None else input_type

multidimensional = (
isinstance(shape, tuple) and len([d for d in shape if d > 1]) > 1
)
multidimensional = isinstance(shape, tuple) and len(shape) > 1
assume(
not (
input_type == "series"
Expand Down

0 comments on commit f2e9459

Please sign in to comment.