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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 3 deletions
Show Stats 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: def _is_cupy_array(data: DataType) -> bool:
return lazy_isinstance(data, "cupy.core.core", "ndarray") or lazy_isinstance( return any(
data, "cupy._core.core", "ndarray" lazy_isinstance(data, n, "ndarray")
for n in ("cupy.core.core", "cupy", "cupy._core.core")
) )

2
tests/ci_build/conda_env/python_lint.yml
View File

@ -6,7 +6,7 @@ dependencies:
- pylint - pylint
- wheel - wheel
- setuptools - setuptools
- mypy - mypy=0.961
- numpy - numpy
- scipy - scipy
- pandas - pandas