From ffc267f94b650c5291fa2ef9b7ea093ccc15bcdc Mon Sep 17 00:00:00 2001 From: Kai Yang Date: Fri, 6 Nov 2020 03:07:34 +0800 Subject: [PATCH] [Test] Ignore setproctitle for local mode (#11819) --- python/ray/worker.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/ray/worker.py b/python/ray/worker.py index 64dadb62c..2ff7f95a2 100644 --- a/python/ray/worker.py +++ b/python/ray/worker.py @@ -1184,7 +1184,7 @@ def connect(node, worker.worker_id = ray.utils.compute_driver_id_from_job( job_id).binary() - if mode is not SCRIPT_MODE and setproctitle: + if mode is not SCRIPT_MODE and mode is not LOCAL_MODE and setproctitle: process_name = ray_constants.WORKER_PROCESS_TYPE_IDLE_WORKER if mode is IO_WORKER_MODE: process_name = ray_constants.WORKER_PROCESS_TYPE_IO_WORKER @@ -1368,11 +1368,13 @@ def disconnect(exiting_interpreter=False): @contextmanager def _changeproctitle(title, next_title): - setproctitle.setproctitle(title) + if _mode() is not LOCAL_MODE: + setproctitle.setproctitle(title) try: yield finally: - setproctitle.setproctitle(next_title) + if _mode() is not LOCAL_MODE: + setproctitle.setproctitle(next_title) def show_in_dashboard(message, key="", dtype="text"):