Skip to content

Commit

Permalink
Lint.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Jan 29, 2021
1 parent 8c34d55 commit 0a417a8
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions python-package/xgboost/dask.py
Original file line number Diff line number Diff line change
Expand Up @@ -1217,8 +1217,9 @@ def predict( # pylint: disable=unused-argument
'''
_assert_dask_support()
client = _xgb_get_client(client)
args = {k: v for k, v in locals().items()}
return client.sync(_predict_async, global_config=config.get_config(), **args)
return client.sync(
_predict_async, global_config=config.get_config(), **locals().items()
)


async def _inplace_predict_async(
Expand Down Expand Up @@ -1309,8 +1310,9 @@ def inplace_predict( # pylint: disable=unused-argument
'''
_assert_dask_support()
client = _xgb_get_client(client)
args = {k: v for k, v in locals().items()}
return client.sync(_inplace_predict_async, global_config=config.get_config(), **args)
return client.sync(
_inplace_predict_async, global_config=config.get_config(), **locals().items()
)


async def _async_wrap_evaluation_matrices(
Expand Down

0 comments on commit 0a417a8

Please sign in to comment.