apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

make sure job is always defined, remove unused placeholders from logs

This commit is contained in:
Sean Sube 2023-07-17 22:46:02 -05:00
parent 01c7392a61
commit c506fe1e9f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/worker/context.py
View File

@ -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

6
api/onnx_web/worker/worker.py
View File

@ -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: