From d45c0d843ba46f1e380293d2926e99144a9e38d0 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Wed, 16 Dec 2020 02:58:21 +0800 Subject: [PATCH] Show partition status in dask error. (#6366) --- python-package/xgboost/dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 = {}