From 140c377a96d5475c4eef3f92b5c784f40035a6e2 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 12 Aug 2022 22:02:05 +0800 Subject: [PATCH] [backport] Fix compatibility with latest cupy. (#8129) (#8160) * Fix compatibility with latest cupy. * Freeze mypy. --- python-package/xgboost/data.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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") )