diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py index 15ef4e515..219737e03 100644 --- a/python-package/xgboost/data.py +++ b/python-package/xgboost/data.py @@ -706,9 +706,10 @@ def _is_cudf_ser(data): return isinstance(data, cudf.Series) -def _is_cupy_array(data: Any) -> bool: - return lazy_isinstance(data, "cupy.core.core", "ndarray") or lazy_isinstance( - data, "cupy._core.core", "ndarray" +def _is_cupy_array(data) -> bool: + return any( + lazy_isinstance(data, n, "ndarray") + for n in ("cupy.core.core", "cupy", "cupy._core.core") )