From 579ab23b10969f473d30c5f18b546963941eccff Mon Sep 17 00:00:00 2001 From: Rafail Giavrimis <47496212+grafail@users.noreply.github.com> Date: Sun, 17 Jul 2022 02:27:43 +0100 Subject: [PATCH] Check cudf lazily (#8084) --- python-package/xgboost/data.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py index a0505e9c9..dbe0a66c8 100644 --- a/python-package/xgboost/data.py +++ b/python-package/xgboost/data.py @@ -745,11 +745,7 @@ def _from_cudf_df( def _is_cudf_ser(data: DataType) -> bool: - try: - import cudf - except ImportError: - return False - return isinstance(data, cudf.Series) + return lazy_isinstance(data, "cudf.core.series", "Series") def _is_cupy_array(data: DataType) -> bool: