perf: exit on error during predict

mbari-org · Sep 19, 2024 · b747deb · b747deb
1 parent a232fd6
commit b747deb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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