Fix compatibility with latest cupy. (#8129)

* Fix compatibility with latest cupy. * Freeze mypy.
dmlc · Aug 1, 2022 · 2cba1d9 · 2cba1d9
1 parent 24c2373
commit 2cba1d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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")
     )
 
 

diff --git 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
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,7 +6,7 @@ dependencies: @@
     - pylint
     - wheel
     - setuptools
-    - mypy
+    - mypy=0.961
     - numpy
     - scipy
     - pandas
@@ Expand Down @@