From b747debeb6e2d234786d397034094ca950ba03cc Mon Sep 17 00:00:00 2001 From: danellecline Date: Thu, 19 Sep 2024 13:51:23 -0700 Subject: [PATCH] perf: exit on error during predict --- aipipeline/prediction/vss_predict_pipeline.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aipipeline/prediction/vss_predict_pipeline.py b/aipipeline/prediction/vss_predict_pipeline.py index c5643e0..e4ea30f 100644 --- a/aipipeline/prediction/vss_predict_pipeline.py +++ b/aipipeline/prediction/vss_predict_pipeline.py @@ -115,6 +115,7 @@ def process_image_batch(batch, config_dict): num_loaded += 1 else: logger.error(f"Error loading label {best_pred} to {database_id} {response.status_code} {response.text}") + exit(1) return num_loaded