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

Fix prediction on df with latest dask. (#6969)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-05-19 12:23:03 +08:00 committed by GitHub
parent 6e104f0570
commit 7e846bb965
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

@ -1000,7 +1000,7 @@ async def _direct_predict_impl( # pylint: disable=too-many-branches
output_shape: Tuple[int, ...], output_shape: Tuple[int, ...],
meta: Dict[int, str], meta: Dict[int, str],
) -> _DaskCollection: ) -> _DaskCollection:
columns = list(meta.keys()) columns = tuple(meta.keys())
if len(output_shape) >= 3 and isinstance(data, dd.DataFrame): if len(output_shape) >= 3 and isinstance(data, dd.DataFrame):
# Without this check, dask will finish the prediction silently even if output # Without this check, dask will finish the prediction silently even if output
# dimension is greater than 3. But during map_partitions, dask passes a # dimension is greater than 3. But during map_partitions, dask passes a