Use array interface for testing numpy arrays.

dmlc · Sep 21, 2023 · 2e1d276 · 2e1d276
1 parent 0080c97
commit 2e1d276
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py
@@ -2434,6 +2434,7 @@ def inplace_predict(
             _is_cudf_df,
             _is_cupy_array,
             _is_list,
+            _is_numpy_array,
             _is_pandas_df,
             _is_pandas_series,
             _is_tuple,
@@ -2463,7 +2464,7 @@ def inplace_predict(
                     f"got {data.shape[1]}"
                 )
 
-        if isinstance(data, np.ndarray):
+        if _is_numpy_array(data):
             from .data import _ensure_np_dtype
 
             data, _ = _ensure_np_dtype(data, data.dtype)

diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py
@@ -165,7 +165,7 @@ def _is_scipy_coo(data: DataType) -> bool:
 
 
 def _is_numpy_array(data: DataType) -> bool:
-    return isinstance(data, (np.ndarray, np.matrix))
+    return hasattr(data, "__array_interface__")
 
 
 def _ensure_np_dtype(