Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Show partition status in dask error. (#6366)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-12-16 02:58:21 +08:00 committed by GitHub
parent 1e2c3ade9e
commit d45c0d843b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/dask.py
View File

@ -312,7 +312,7 @@ class DaskDMatrix:
await distributed.wait(parts) # async wait for parts to be computed await distributed.wait(parts) # async wait for parts to be computed
for part in parts: for part in parts:
assert part.status == 'finished' assert part.status == 'finished', part.status
# Preserving the partition order for prediction. # Preserving the partition order for prediction.
self.partition_order = {} self.partition_order = {}