diff --git a/python-package/xgboost/spark/utils.py b/python-package/xgboost/spark/utils.py index 7dbe290ae3e5..1403596c0c2a 100644 --- a/python-package/xgboost/spark/utils.py +++ b/python-package/xgboost/spark/utils.py @@ -55,7 +55,7 @@ def _start_tracker(context: BarrierTaskContext, n_workers: int) -> Dict[str, Any """Start Rabit tracker with n_workers""" env: Dict[str, Any] = {"DMLC_NUM_WORKER": n_workers} host = _get_host_ip(context) - rabit_context = RabitTracker(host_ip=host, n_workers=n_workers) + rabit_context = RabitTracker(host_ip=host, n_workers=n_workers, sortby="task") env.update(rabit_context.worker_envs()) rabit_context.start(n_workers) thread = Thread(target=rabit_context.join)