apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-03-26 14:22:59 -05:00
parent 0af406c47f
commit 27500eccb5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/worker/pool.py
View File

@ -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)