From d8d2fd25cd15cf05cf6d4ec6560168fc0b058fe5 Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Mon, 10 Jun 2024 16:02:50 -0700 Subject: [PATCH] predict: fix batch error reporting --- plugins/openvino/src/predict/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/plugins/openvino/src/predict/__init__.py b/plugins/openvino/src/predict/__init__.py index 33b092b1d..202a759e6 100644 --- a/plugins/openvino/src/predict/__init__.py +++ b/plugins/openvino/src/predict/__init__.py @@ -145,8 +145,8 @@ class PredictPlugin(DetectPlugin): 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