diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index bdc0449a..9f04dfbf 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -146,7 +146,7 @@ class DevicePoolExecutor: queued = job_counts.most_common() logger.debug('jobs queued by device: %s', queued) - return queued[-1] + return queued[-1][0] def prune(self): self.jobs[:] = [job for job in self.jobs if job.future.done()]