From de6e3b4e27f28294c9d30969a471ceafd5af9915 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 16:40:18 -0600 Subject: [PATCH] only return device index --- api/onnx_web/device_pool.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()]