From c262a3c8ffaeefdf441261281a058aeced1056ae Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Thu, 31 Oct 2024 23:30:45 -0700 Subject: [PATCH] Fix dmlc/xgboost#10752 --- tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py b/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py index 982a68c05ba3..5e4ada563259 100644 --- a/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py +++ b/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py @@ -97,6 +97,8 @@ def is_df(part: T) -> T: cp.testing.assert_allclose(predt.values.compute(), single_node) # Make sure the output can be integrated back to original dataframe + X.columns = X.columns.astype("object") + # Work around https://github.com/dmlc/xgboost/issues/10752 X["predict"] = predictions X["inplace_predict"] = series_predictions