From 66c2ce3074f8af4656848ba562a7c4bad3135208 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 11 Feb 2023 16:19:08 -0600 Subject: [PATCH] fix device iteration --- 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 c7cd6395..ff6ca4c2 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -159,7 +159,7 @@ class DevicePoolExecutor: def get_next_device(self, needs_device: Optional[DeviceParams] = None) -> int: # respect overrides if possible if needs_device is not None: - for i in self.devices: + for i in range(len(self.devices)): if self.devices[i].device == needs_device.device: return i