apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix finished flag

This commit is contained in:
Sean Sube 2023-02-26 20:13:16 -06:00
parent 525ee24e91
commit 401ee20526
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -134,7 +134,7 @@ class DevicePoolExecutor:
def done(self, key: str) -> Tuple[Optional[bool], int]: def done(self, key: str) -> Tuple[Optional[bool], int]:
for k, p, c in self.finished_jobs: for k, p, c in self.finished_jobs:
if k == key: if k == key:
return (c, p) return (True, p)
if key not in self.active_jobs: if key not in self.active_jobs:
logger.warn("checking status for unknown job: %s", key) logger.warn("checking status for unknown job: %s", key)