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)