further testing changes

uxlfoundation · Nov 20, 2024 · 27050bd · 27050bd
1 parent 740a5e7
commit 27050bd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sklearnex/utils/tests/test_finite.py b/sklearnex/utils/tests/test_finite.py
@@ -65,7 +65,7 @@ def test_assert_finite_random_location(dtype, shape, allow_nan, check, seed):
         X.reshape((-1,))[loc] = float(check)
 
     if check is None or (allow_nan and check == "NaN"):
-        _assert_all_finite(X, allow_nan=allow_nan)
+        assert_all_finite(X, allow_nan=allow_nan)
     else:
         assert_raises(ValueError, _assert_all_finite, X, allow_nan=allow_nan)
 
@@ -84,6 +84,6 @@ def test_assert_finite_random_shape_and_location(dtype, allow_nan, check, seed):
         X[loc] = float(check)
 
     if check is None or (allow_nan and check == "NaN"):
-        _assert_all_finite(X, allow_nan=allow_nan)
+        assert_all_finite(X, allow_nan=allow_nan)
     else:
-        assert_raises(ValueError, _assert_all_finite, X, allow_nan=allow_nan)
+        assert_raises(ValueError, assert_all_finite, X, allow_nan=allow_nan)