predict: fix batch error reporting

koush · Jun 10, 2024 · d8d2fd2 · d8d2fd2
1 parent 301a5b6
commit d8d2fd2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/openvino/src/predict/__init__.py b/plugins/openvino/src/predict/__init__.py
@@ -145,8 +145,8 @@ async def run_batch(self):
                 for i, result in enumerate(results):
                     batch[i][1].set_result(result)
             except Exception as e:
-                for i, result in enumerate(results):
-                    batch[i][1].set_exception(e)
+                for input in batch:
+                    input[1].set_exception(e)
 
     async def flush_batch(self):
         self.batch_flush = None