[dask] Remove the workaround for segfault. (#7146)

2021-07-30 03:57:53 +08:00 · 2021-07-30 03:57:53 +08:00 · 1369133916
commit 1369133916
parent f1a4a1ac95
1 changed files with 1 additions and 2 deletions
--- a/python-package/xgboost/dask.py
+++ b/python-package/xgboost/dask.py
@ -1219,8 +1219,7 @@ async def _predict_async(
        base_margin = None
        for i, blob in enumerate(part[1:]):
            if meta_names[i] == "base_margin":
-                # segfault without copy. See https://github.com/dmlc/xgboost/issues/7111.
-                base_margin = blob.copy()
+                base_margin = blob
        with config.config_context(**global_config):
            m = DMatrix(
                data,