fix(api): do not move jobs from pending to running until progress is …

…reported
ssube · Mar 26, 2023 · 27500ec · 27500ec
1 parent 0af406c
commit 27500ec
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py
@@ -195,9 +195,7 @@ def cancel(self, key: str) -> bool:
 
         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 @@ def next_job(self, device: str):
             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)