From ae434f5c11fa1fda188901cd4e7338179fef8bb9 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 16:55:20 -0600 Subject: [PATCH] fix(api): add device index to job status endpoint --- api/onnx_web/device_pool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 825fa68d..c6ea707e 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -169,5 +169,5 @@ class DevicePoolExecutor: future.add_done_callback(job_done) - def status(self) -> Dict[str, Tuple[bool, int]]: - return [(job.key, job.future.done(), job.get_progress()) for job in self.jobs] + def status(self) -> List[Tuple[str, int, bool, int]]: + return [(job.key, job.context.device_index.value, job.future.done(), job.get_progress()) for job in self.jobs]