apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

always put progress in active jobs

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

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

@ -122,7 +122,7 @@ class DevicePoolExecutor:
logger.warn("attempting to cancel unknown job: %s", key)
return False
device = self.active_jobs[key]
device, _progress = self.active_jobs[key]
context = self.context[device]
logger.info("cancelling job %s on device %s", key, device)
@ -220,7 +220,7 @@ class DevicePoolExecutor:
queue = self.pending[device.device]
queue.put((fn, args, kwargs))
self.active_jobs[key] = device.device
self.active_jobs[key] = (device.device, 0)
def status(self) -> List[Tuple[str, int, bool, int]]:
pending = [
@ -232,7 +232,7 @@ class DevicePoolExecutor:
False,
progress,
)
for name, device, progress in self.active_jobs
for name, device, progress in self.active_jobs.items()
]
pending.extend(
[