Add an API guard to prevent global variables being changed. (#6891)

2021-04-23 10:27:57 +08:00
parent 896aede340
commit a2ecbdaa31
7 changed files with 96 additions and 24 deletions
--- a/python-package/xgboost/core.py
+++ b/python-package/xgboost/core.py
@@ -1939,8 +1939,11 @@ class Booster(object):
                )
            )
            return _prediction_output(shape, dims, preds, False)
-        if lazy_isinstance(data, "cupy.core.core", "ndarray"):
+        if lazy_isinstance(data, "cupy.core.core", "ndarray") or lazy_isinstance(
+            data, "cupy._core.core", "ndarray"
+        ):
            from .data import _transform_cupy_array
+
            data = _transform_cupy_array(data)
            interface = data.__cuda_array_interface__
            if "mask" in interface: