Fix dmlc#10752

hcho3 · Nov 1, 2024 · c262a3c · c262a3c
1 parent 622a4ab
commit c262a3c
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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