Fix test

dmlc · Apr 13, 2024 · fc12745 · fc12745
1 parent 7e1d1a9
commit fc12745
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/python-package/xgboost/data.py b/python-package/xgboost/data.py
@@ -915,8 +915,9 @@ def _transform_cudf_df(
         from pandas.api.types import is_bool_dtype
 
     # Work around https://github.com/dmlc/xgboost/issues/10181
-    if _is_cudf_ser(data) and is_bool_dtype(data.dtype):
-        data = data.astype(np.uint8)
+    if _is_cudf_ser(data):
+        if is_bool_dtype(data.dtype):
+            data = data.astype(np.uint8)
     else:
         data = data.astype(
             {col: np.uint8 for col in data.select_dtypes(include="bool")}

diff --git a/tests/python-gpu/test_from_cudf.py b/tests/python-gpu/test_from_cudf.py
@@ -71,15 +71,6 @@ def _test_from_cudf(DMatrixT):
     assert dtrain.num_col() == 1
     assert dtrain.num_row() == 5
 
-    # Boolean is not supported.
-    X_boolean = cudf.DataFrame({"x": cudf.Series([True, False])})
-    with pytest.raises(Exception):
-        dtrain = DMatrixT(X_boolean)
-
-    y_boolean = cudf.DataFrame({"x": cudf.Series([True, False, True, True, True])})
-    with pytest.raises(Exception):
-        dtrain = DMatrixT(X_boolean, label=y_boolean)
-
 
 def _test_cudf_training(DMatrixT):
     import pandas as pd