From 27500eccb570d78004f97062ddb4404f2c0a8016 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 26 Mar 2023 14:22:59 -0500 Subject: [PATCH] fix(api): do not move jobs from pending to running until progress is reported --- api/onnx_web/worker/pool.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py index 953aba9a..a065d829 100644 --- a/api/onnx_web/worker/pool.py +++ b/api/onnx_web/worker/pool.py @@ -195,9 +195,7 @@ class DevicePoolExecutor: for job in self.pending_jobs: if job.name == key: - self.pending_jobs[:] = [ - job for job in self.pending_jobs if job.name != key - ] + self.pending_jobs.remove(job) logger.info("cancelled pending job: %s", key) return True @@ -414,7 +412,7 @@ class DevicePoolExecutor: if job.device == device: logger.debug("enqueuing job %s on device %s", job.name, device) self.pending[device].put(job, block=False) - self.pending_jobs.remove(job) + # job will be removed from pending queue when progress is updated return logger.trace("no pending jobs for device %s", device)