From 87bbce4fffc66c2041dbaf72129cb5941a69ab35 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 11:08:22 -0600 Subject: [PATCH] return progress through chain --- api/onnx_web/device_pool.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 59189652..d142556f 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -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] \ No newline at end of file + return [(job.key, job.future.done(), job.get_progress()) for job in self.jobs] \ No newline at end of file