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

[backport] Fix compatibility with latest cupy. (#8129) (#8160)

Browse Source 
* Fix compatibility with latest cupy.

* Freeze mypy.
This commit is contained in:
Jiaming Yuan 2022-08-12 22:02:05 +08:00 committed by GitHub
parent 39c1488a42
commit 140c377a96
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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")
)