From e08a9aa2abaacd7066c0223b6e052eb4f78286f3 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Mar 2023 17:26:28 -0500 Subject: [PATCH] add pending job list to pool --- api/onnx_web/worker/pool.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py index bbcdb726..d50fede4 100644 --- a/api/onnx_web/worker/pool.py +++ b/api/onnx_web/worker/pool.py @@ -58,9 +58,10 @@ class DevicePoolExecutor: self.threads = {} self.workers = {} - self.running_jobs = {} self.cancelled_jobs = [] self.finished_jobs = [] + self.pending_jobs = [] + self.running_jobs = {} self.total_jobs = {} self.logs = Queue(self.max_pending_per_worker)