apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): add device index to job status endpoint

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

4
api/onnx_web/device_pool.py
View File

@ -169,5 +169,5 @@ class DevicePoolExecutor:
future.add_done_callback(job_done) future.add_done_callback(job_done)
def status(self) -> Dict[str, Tuple[bool, int]]: def status(self) -> List[Tuple[str, int, bool, int]]:
return [(job.key, job.future.done(), job.get_progress()) for job in self.jobs] return [(job.key, job.context.device_index.value, job.future.done(), job.get_progress()) for job in self.jobs]