From c393d628eb3f695af50f511661544601f5ea730d Mon Sep 17 00:00:00 2001
From: Simon Adorf <sadorf@nvidia.com>
Date: Tue, 14 Nov 2023 09:22:08 -0800
Subject: [PATCH] Adjust assumption regarding valid cudf.Series dimensional
 input.

cudf.Series does not accept any multi-dimensional input anymore.
---
 python/cuml/testing/strategies.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/python/cuml/testing/strategies.py b/python/cuml/testing/strategies.py
index 39b2068e5d..a8f849866a 100644
--- a/python/cuml/testing/strategies.py
+++ b/python/cuml/testing/strategies.py
@@ -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"