From 7a6de9630dcd7be077826ec7edfbf7417722a484 Mon Sep 17 00:00:00 2001 From: fis Date: Tue, 16 Nov 2021 03:33:57 +0800 Subject: [PATCH] Set use_logger in tracker to false. --- 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 4594f37e0517..f2fc90302c0e 100644 --- a/python-package/xgboost/tracker.py +++ b/python-package/xgboost/tracker.py @@ -162,7 +162,7 @@ def __init__( n_workers: int, port: int = 9091, port_end: int = 9999, - use_logger: bool = True, + use_logger: bool = False, ) -> None: """A Python implementation of RABIT tracker. @@ -407,16 +407,17 @@ def start_rabit_tracker(args: argparse.Namespace) -> None: ---------- 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), n_workers=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), n_workers=args.num_workers, use_logger=True + ) envs.update(rabit.worker_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()