From 9bbd00a49f861b913d989fb638eb83cdf73be9c3 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Tue, 16 Nov 2021 09:51:37 +0800 Subject: [PATCH] [backport] Set use_logger in tracker to false. (#7438) (#7439) --- python-package/xgboost/tracker.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/python-package/xgboost/tracker.py b/python-package/xgboost/tracker.py index c64e0486051f..c90d2839c3d3 100644 --- a/python-package/xgboost/tracker.py +++ b/python-package/xgboost/tracker.py @@ -144,7 +144,7 @@ class RabitTracker(object): """ def __init__( - self, hostIP, nslave, port=9091, port_end=9999, use_logger: bool = True + self, hostIP, nslave, port=9091, port_end=9999, use_logger: bool = False ) -> None: """A Python implementation of RABIT tracker. @@ -384,16 +384,17 @@ def start_rabit_tracker(args): ---------- args: arguments to start the rabit tracker. """ - envs = {'DMLC_NUM_WORKER': args.num_workers, - 'DMLC_NUM_SERVER': args.num_servers} - rabit = RabitTracker(hostIP=get_host_ip(args.host_ip), nslave=args.num_workers) + envs = {"DMLC_NUM_WORKER": args.num_workers, "DMLC_NUM_SERVER": args.num_servers} + rabit = RabitTracker( + hostIP=get_host_ip(args.host_ip), nslave=args.num_workers, use_logger=True + ) envs.update(rabit.slave_envs()) rabit.start(args.num_workers) - sys.stdout.write('DMLC_TRACKER_ENV_START\n') + sys.stdout.write("DMLC_TRACKER_ENV_START\n") # simply write configuration to stdout for k, v in envs.items(): sys.stdout.write(f"{k}={v}\n") - sys.stdout.write('DMLC_TRACKER_ENV_END\n') + sys.stdout.write("DMLC_TRACKER_ENV_END\n") sys.stdout.flush() rabit.join()