michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-06-12 13:40:24 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

predict: fix batch error reporting

Browse Source
This commit is contained in:
Koushik Dutta
2024-06-10 16:02:50 -07:00
parent 301a5b6685
commit d8d2fd25cd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/openvino/src/predict/__init__.py
View File

@@ -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