From ada4a86d1c3687ba78392993e6b2902d69175ef2 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 3 Sep 2022 03:10:43 +0800 Subject: [PATCH] Fix dask interface with latest cupy. (#8210) --- python-package/xgboost/compat.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/compat.py b/python-package/xgboost/compat.py index 1ef1bbaab..275b66210 100644 --- a/python-package/xgboost/compat.py +++ b/python-package/xgboost/compat.py @@ -4,7 +4,7 @@ import importlib.util import logging import sys import types -from typing import Any, Dict, List, Optional, Sequence, Type, cast +from typing import Any, Dict, List, Optional, Sequence, cast import numpy as np @@ -125,7 +125,9 @@ def concat(value: Sequence[_T]) -> _T: # pylint: disable=too-many-return-statem 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=import-error # pylint: disable=c-extension-no-member,no-member