fix(api): only remove running jobs from running state

ssube · Mar 19, 2023 · 2e89fd4 · 2e89fd4
1 parent 716e009
commit 2e89fd4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py
@@ -413,7 +413,9 @@ def update_job(self, progress: ProgressCommand):
             # move from running to finished
             logger.info("job has finished: %s", progress.job)
             self.finished_jobs.append(progress)
-            del self.running_jobs[progress.job]
+            if progress.job in self.running_jobs:
+                del self.running_jobs[progress.job]
+
             self.join_leaking()
             if progress.job in self.cancelled_jobs:
                 self.cancelled_jobs.remove(progress.job)