Fix dask predict on DaskDMatrix with iteration_range. (#7005)

2021-05-29 04:43:12 +08:00
parent 4cf95a6041
commit 89a49cf30e
2 changed files with 35 additions and 0 deletions
--- a/python-package/xgboost/dask.py
+++ b/python-package/xgboost/dask.py
@@ -1217,6 +1217,8 @@ async def _predict_async(
                approx_contribs=approx_contribs,
                pred_interactions=pred_interactions,
                validate_features=validate_features,
+                iteration_range=iteration_range,
+                strict_shape=strict_shape,
            )
            return predt