From e3fbdbd7d755f9472d446782454172d074941ee4 Mon Sep 17 00:00:00 2001 From: danellecline Date: Fri, 18 Oct 2024 13:59:14 -0700 Subject: [PATCH] fix: continue to load exemplar when missing data as some labels may be missing --- aipipeline/prediction/vss_load_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aipipeline/prediction/vss_load_pipeline.py b/aipipeline/prediction/vss_load_pipeline.py index 731f95a..3d28d1e 100755 --- a/aipipeline/prediction/vss_load_pipeline.py +++ b/aipipeline/prediction/vss_load_pipeline.py @@ -71,7 +71,7 @@ def load_exemplars(labels: List[tuple[str, str]], config_dict=Dict, conf_files=D if exemplar_file is None: logger.info(f"No detections file found for {label}") - return f"No exemplar or detections file found for {label}" + continue with open(exemplar_file, "r") as f: exemplar_count = len(f.readlines())