From 2e89fd43d3223177bc9a157a9b9a8fa1a06c5638 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Mar 2023 19:21:40 -0500 Subject: [PATCH] fix(api): only remove running jobs from running state --- api/onnx_web/worker/pool.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/worker/pool.py b/api/onnx_web/worker/pool.py index 31bd0df1..b704767f 100644 --- a/api/onnx_web/worker/pool.py +++ b/api/onnx_web/worker/pool.py @@ -413,7 +413,9 @@ class DevicePoolExecutor: # 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)