Show partition status in dask error. (#6366)

2020-12-16 02:58:21 +08:00
parent 1e2c3ade9e
commit d45c0d843b
1 changed files with 1 additions and 1 deletions
--- 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 = {}