Fix compatibility with latest cupy. (#8129)

* Fix compatibility with latest cupy. * Freeze mypy.
2022-08-01 15:24:42 +08:00 · 2022-08-01 15:24:42 +08:00 · 2cba1d9fcc
commit 2cba1d9fcc
parent 24c2373080
2 changed files with 4 additions and 3 deletions
--- a/python-package/xgboost/data.py
+++ b/python-package/xgboost/data.py
@ -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")
    )


--- a/tests/ci_build/conda_env/python_lint.yml
+++ b/tests/ci_build/conda_env/python_lint.yml
@ -6,7 +6,7 @@ dependencies:
 - pylint
 - wheel
 - setuptools
- mypy
+- mypy=0.961
 - numpy
 - scipy
 - pandas