Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
hcho3 committed Apr 13, 2024
1 parent 2674b67 commit 3643a96
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python-package/xgboost/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -918,7 +918,9 @@ def _transform_cudf_df(
if _is_cudf_ser(data) and 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")})
data = data.astype(
{col: np.uint8 for col in data.select_dtypes(include="bool")}
)

if _is_cudf_ser(data):
dtypes = [data.dtype]
Expand Down

0 comments on commit 3643a96

Please sign in to comment.