[Test] Ignore setproctitle for local mode (#11819)

This commit is contained in:
Kai Yang
2020-11-06 03:07:34 +08:00
committed by Alex Wu
parent 6e6cb6b1d0
commit 2a86943b13
+5 -3
View File
@@ -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"):