[dask] Fix nthread config with dask sklearn wrapper. (#7633)

2022-02-08 06:38:32 +08:00
parent d152c59a9c
commit 3e693e4f97
2 changed files with 8 additions and 4 deletions
--- a/python-package/xgboost/dask.py
+++ b/python-package/xgboost/dask.py
@@ -944,7 +944,7 @@ async def _train_async(
                LOGGER.info("Overriding `nthreads` defined in dask worker.")
                n_threads = local_param[p]
                break
-        if n_threads == 0:
+        if n_threads == 0 or n_threads is None:
            n_threads = worker.nthreads
        local_param.update({"nthread": n_threads, "n_jobs": n_threads})