From f83f1ef1a3217a1997434c09fbe8efc6777ecdc8 Mon Sep 17 00:00:00 2001
From: "Faust, Ian" <ian.faust@intel.com>
Date: Wed, 27 Nov 2024 11:59:43 +0100
Subject: [PATCH] follow changes made in #2126

---
 onedal/utils/tests/test_validation.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/onedal/utils/tests/test_validation.py b/onedal/utils/tests/test_validation.py
index 37486f0337..31eb8da2cc 100644
--- a/onedal/utils/tests/test_validation.py
+++ b/onedal/utils/tests/test_validation.py
@@ -33,9 +33,7 @@
     "shape",
     [
         [16, 2048],
-        [
-            2**16 + 3,
-        ],
+        [65539],  # 2**16 + 3,
         [1000, 1000],
         [
             3,
@@ -58,9 +56,7 @@ def test_sum_infinite_actually_finite(dtype, shape, allow_nan, dataframe, queue)
     "shape",
     [
         [16, 2048],
-        [
-            2**16 + 3,
-        ],
+        [65539],  # 2**16 + 3,
         [1000, 1000],
         [
             3,