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