diff --git a/api/onnx_web/worker/context.py b/api/onnx_web/worker/context.py index 97aa8acb..9ab7b598 100644 --- a/api/onnx_web/worker/context.py +++ b/api/onnx_web/worker/context.py @@ -36,6 +36,7 @@ class WorkerContext: active_pid: "Value[int]", idle: "Value[bool]", ): + self.job = None self.name = name self.device = device self.cancel = cancel diff --git a/api/onnx_web/worker/worker.py b/api/onnx_web/worker/worker.py index 8d714722..361c1150 100644 --- a/api/onnx_web/worker/worker.py +++ b/api/onnx_web/worker/worker.py @@ -67,15 +67,15 @@ def worker_main(worker: WorkerContext, server: ServerContext): logger.trace("worker reached end of queue, setting idle flag") worker.set_idle() except KeyboardInterrupt: - logger.info("worker got keyboard interrupt") + logger.debug("worker got keyboard interrupt") worker.fail() exit(EXIT_INTERRUPT) except RetryException: - logger.info("retry error in worker, exiting: %s") + logger.exception("retry error in worker, exiting") worker.fail() exit(EXIT_ERROR) except ValueError: - logger.exception("value error in worker, exiting: %s") + logger.exception("value error in worker, exiting") worker.fail() exit(EXIT_ERROR) except Exception as e: