[dask] Use nthread in DMatrix construction. (#7337)

This is consistent with the thread overriding behavior.
2021-10-20 15:16:40 +08:00
parent b8e8f0fcd9
commit f999897615
3 changed files with 44 additions and 33 deletions
--- a/tests/python-gpu/test_gpu_with_dask.py
+++ b/tests/python-gpu/test_gpu_with_dask.py
@@ -434,7 +434,7 @@ class TestDistributedGPU:

            def worker_fn(worker_addr: str, data_ref: Dict) -> None:
                with dxgb.RabitContext(rabit_args):
-                    local_dtrain = dxgb._dmatrix_from_list_of_parts(**data_ref)
+                    local_dtrain = dxgb._dmatrix_from_list_of_parts(**data_ref, nthread=7)
                    fw_rows = local_dtrain.get_float_info("feature_weights").shape[0]
                    assert fw_rows == local_dtrain.num_col()