follow changes made in uxlfoundation#2126

icfaust · Nov 27, 2024 · f83f1ef · f83f1ef
1 parent c4dccd6
commit f83f1ef
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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,