From 2a86943b13983d677fb2b1b27b25d7096fa5feb3 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 06bf9d66d..888dc8569 100644 --- a/python/ray/worker.py +++ b/python/ray/worker.py @@ -1164,7 +1164,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 @@ -1348,11 +1348,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"):