move _raise_deadline() out of except block to clean up printed stacktrace

1 files changed, 4 insertions, 0 deletions

diff --git a/src/utils/__init__.py b/src/utils/__init__.py
index 8f4920f4..60f25f9d 100644
--- a/src/utils/__init__.py
+++ b/src/utils/__init__.py
@@ -398,10 +398,14 @@ def deadline_process(func: typing.Callable[[], None], seconds: int=10):
     p = multiprocessing.Process(target=_wrap, args=(func, q))
     p.start()
 
+    deadlined = False
     try:
         success, out = q.get(block=True, timeout=seconds)
     except queue.Empty:
         p.kill()
+        deadlined = True
+
+    if deadlined:
         _raise_deadline()
 
     if success: