From edc55ae8b49d11e20f4fd69a05ab22de5f6f74e4 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Mar 2023 19:53:44 -0600 Subject: [PATCH] fix(api): finished job notification should not block worker --- api/onnx_web/worker/context.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/worker/context.py b/api/onnx_web/worker/context.py index f9eb25b3..83f2da1d 100644 --- a/api/onnx_web/worker/context.py +++ b/api/onnx_web/worker/context.py @@ -66,7 +66,7 @@ class WorkerContext: self.progress.put((self.job, self.device.device, progress), block=False) def set_finished(self) -> None: - self.finished.put((self.job, self.device.device)) + self.finished.put((self.job, self.device.device), block=False) def clear_flags(self) -> None: self.set_cancel(False)