Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix compatibility with latest cupy. (#8129)

Browse Source 
* Fix compatibility with latest cupy.

* Freeze mypy.
This commit is contained in:
Jiaming Yuan
2022-08-01 15:24:42 +08:00
committed by GitHub
parent 24c2373080
commit 2cba1d9fcc
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python-package/xgboost/data.py
View File

@@ -749,8 +749,9 @@ def _is_cudf_ser(data: DataType) -> bool:
def _is_cupy_array(data: DataType) -> bool:
return lazy_isinstance(data, "cupy.core.core", "ndarray") or lazy_isinstance(
data, "cupy._core.core", "ndarray"
return any(
lazy_isinstance(data, n, "ndarray")
for n in ("cupy.core.core", "cupy", "cupy._core.core")
)