From 988088d64efdef3af8c576e9ee96d1ed5f5e777d Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 27 Feb 2023 22:52:43 -0600 Subject: [PATCH] quit workers on keyboard signal --- api/onnx_web/worker/worker.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/onnx_web/worker/worker.py b/api/onnx_web/worker/worker.py index ff0e9aae..691a0fad 100644 --- a/api/onnx_web/worker/worker.py +++ b/api/onnx_web/worker/worker.py @@ -1,5 +1,6 @@ from logging import getLogger from queue import Empty +from sys import exit from traceback import format_exception from setproctitle import setproctitle @@ -31,6 +32,8 @@ def worker_main(context: WorkerContext, server: ServerContext): context.set_finished() except Empty: pass + except KeyboardInterrupt: + exit(0) except Exception as e: logger.error( "error while running job: %s",