diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py index c6a0d96c5e80..6fa26b657d81 100644 --- a/python-package/xgboost/dask.py +++ b/python-package/xgboost/dask.py @@ -272,7 +272,9 @@ def concat(value: Any) -> Any: # pylint: disable=too-many-return-statements from cudf import concat as CUDF_concat # pylint: disable=import-error return CUDF_concat(value, axis=0) - if lazy_isinstance(value[0], "cupy._core.core", "ndarray"): + from .data import _is_cupy_array + + if _is_cupy_array(value[0]): import cupy # pylint: disable=c-extension-no-member,no-member diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py index 219737e036c5..a9af782db1c0 100644 --- a/python-package/xgboost/data.py +++ b/python-package/xgboost/data.py @@ -712,7 +712,6 @@ def _is_cupy_array(data) -> bool: for n in ("cupy.core.core", "cupy", "cupy._core.core") ) - def _transform_cupy_array(data): import cupy # pylint: disable=import-error if not hasattr(data, '__cuda_array_interface__') and hasattr(