Check cudf lazily (#8084)

dmlc · Jul 17, 2022 · 579ab23 · 579ab23
1 parent a33f35e
commit 579ab23
Showing 1 changed file with 1 addition and 5 deletions.
diff --git 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: