From 44825d81e914dd1fdb89fe47f9bac3affe075fbd Mon Sep 17 00:00:00 2001 From: ijrsvt Date: Wed, 8 Apr 2020 11:33:43 -0700 Subject: [PATCH] Change Proctitle to IDLE after an Error (#7863) --- python/ray/_raylet.pyx | 4 ++-- python/ray/worker.py | 6 ++++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx index 45b7a42c2..31e74d820 100644 --- a/python/ray/_raylet.pyx +++ b/python/ray/_raylet.pyx @@ -473,10 +473,10 @@ cdef execute_task( if isinstance(error, RayTaskError): # Avoid recursive nesting of RayTaskError. failure_object = RayTaskError(function_name, backtrace, - error.cause_cls) + error.cause_cls, proctitle=title) else: failure_object = RayTaskError(function_name, backtrace, - error.__class__) + error.__class__, proctitle=title) errors = [] for _ in range(c_return_ids.size()): errors.append(failure_object) diff --git a/python/ray/worker.py b/python/ray/worker.py index 18313d91a..e699b7230 100644 --- a/python/ray/worker.py +++ b/python/ray/worker.py @@ -1344,8 +1344,10 @@ def disconnect(exiting_interpreter=False): @contextmanager def _changeproctitle(title, next_title): setproctitle.setproctitle(title) - yield - setproctitle.setproctitle(next_title) + try: + yield + finally: + setproctitle.setproctitle(next_title) def register_custom_serializer(cls,