diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py index 67bb0d855..0a1a20c39 100644 --- a/python-package/xgboost/dask.py +++ b/python-package/xgboost/dask.py @@ -312,7 +312,7 @@ class DaskDMatrix: await distributed.wait(parts) # async wait for parts to be computed for part in parts: - assert part.status == 'finished' + assert part.status == 'finished', part.status # Preserving the partition order for prediction. self.partition_order = {}