Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Dispatch thrust versions and upgrade rmm. (#7254)

Browse Source 
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan
2021-09-25 03:43:23 +08:00
committed by GitHub
parent fbd58bf190
commit ca17f8a5fc
6 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python-package/xgboost/data.py
View File

@@ -435,7 +435,11 @@ def _cudf_array_interfaces(data) -> Tuple[list, bytes]:
interface is finished.
"""
from cudf.utils.dtypes import is_categorical_dtype
try:
from cudf.api.types import is_categorical_dtype
except ImportError:
from cudf.utils.dtypes import is_categorical_dtype
cat_codes = []
interfaces = []
if _is_cudf_ser(data):
@@ -461,7 +465,10 @@ def _transform_cudf_df(
feature_types: Optional[List[str]],
enable_categorical: bool,
):
from cudf.utils.dtypes import is_categorical_dtype
try:
from cudf.api.types import is_categorical_dtype
except ImportError:
from cudf.utils.dtypes import is_categorical_dtype
if feature_names is None:
if _is_cudf_ser(data):