File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
python-package/xgboost/dask Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -1237,12 +1237,14 @@ def _infer_predict_output(
1237
1237
async def _get_model_future (
1238
1238
client : "distributed.Client" , model : Union [Booster , Dict , "distributed.Future" ]
1239
1239
) -> "distributed.Future" :
1240
- # See https://github.com/dask/dask/issues/11179#issuecomment-2168094529 for
1241
- # the use of hash.
1240
+ # See https://github.com/dask/dask/issues/11179#issuecomment-2168094529 for the use
1241
+ # of hash.
1242
+ # https://github.com/dask/distributed/pull/8796 Don't use broadcast in the `scatter`
1243
+ # call, otherwise, the predict function might hang.
1242
1244
if isinstance (model , Booster ):
1243
- booster = await client .scatter (model , broadcast = True , hash = False )
1245
+ booster = await client .scatter (model , hash = False )
1244
1246
elif isinstance (model , dict ):
1245
- booster = await client .scatter (model ["booster" ], broadcast = True , hash = False )
1247
+ booster = await client .scatter (model ["booster" ], hash = False )
1246
1248
elif isinstance (model , distributed .Future ):
1247
1249
booster = model
1248
1250
t = booster .type
You can’t perform that action at this time.
0 commit comments