FIX avoid warning in input validation with array_api_strict (scikit-learn#29086)

ogrisel · web-flow · commit 60e4acd50d65 · 2024-05-23T16:01:15.000-04:00
diff --git a/sklearn/utils/validation.py b/sklearn/utils/validation.py
@@ -95,15 +95,15 @@ def _assert_all_finite(
 ):
     """Like assert_all_finite, but only for ndarray."""
 
-    xp, _ = get_namespace(X)
+    xp, is_array_api = get_namespace(X)
 
     if _get_config()["assume_finite"]:
         return
 
     X = xp.asarray(X)
 
     # for object dtype data, we only check for NaNs (GH-13254)
-    if X.dtype == np.dtype("object") and not allow_nan:
+    if not is_array_api and X.dtype == np.dtype("object") and not allow_nan:
         if _object_dtype_isnan(X).any():
             raise ValueError("Input contains NaN")
 
