Fix a unit test when cudf is missing

dmlc · Aug 8, 2020 · 377580a · 377580a
1 parent 9021a75
commit 377580a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py
@@ -317,7 +317,7 @@ def _is_cudf_df(data):
         import cudf
     except ImportError:
         return False
-    return isinstance(data, cudf.DataFrame)
+    return hasattr(cudf, 'DataFrame') and isinstance(data, cudf.DataFrame)
 
 
 def _cudf_array_interfaces(data):