diff --git a/sklearnex/utils/tests/test_validation.py b/sklearnex/utils/tests/test_validation.py index 92ba0d742a..70da28dbce 100644 --- a/sklearnex/utils/tests/test_validation.py +++ b/sklearnex/utils/tests/test_validation.py @@ -42,9 +42,7 @@ "shape", [ [16, 2048], - [ - 2**16 + 3, - ], + [2**16 + 3], [1000, 1000], ], ) @@ -63,9 +61,7 @@ def test_sum_infinite_actually_finite(dtype, shape, ensure_all_finite): "shape", [ [16, 2048], - [ - 2**16 + 3, - ], + [2**16 + 3], [1000, 1000], ], )