apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

return progress through chain

This commit is contained in:
Sean Sube 2023-02-04 11:08:22 -06:00
parent 157ed6da70
commit 87bbce4fff
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/onnx_web/device_pool.py
View File

@ -60,6 +60,9 @@ class JobContext:
class Job:
'''
Link a future to its context.
'''
def __init__(
self,
key: str,
@ -71,13 +74,13 @@ class Job:
self.key = key
def get_progress(self) -> int:
self.context.get_progress()
return self.context.get_progress()
def set_cancel(self, cancel: bool = True):
self.context.set_cancel(cancel)
return self.context.set_cancel(cancel)
def set_progress(self, progress: int):
self.context.set_progress(progress)
return self.context.set_progress(progress)
class DevicePoolExecutor:
@ -123,4 +126,4 @@ class DevicePoolExecutor:
self.jobs.append(job)
def status(self) -> Dict[str, Tuple[bool, int]]:
return [(job.future.done(), job.get_progress()) for job in self.jobs]
return [(job.key, job.future.done(), job.get_progress()) for job in self.jobs]