Merge pull request #260 from TogetherCrew/feat/218-violation-detection

fix: skipping None labels for reporting!
TogetherCrew · Aug 12, 2024 · 370989b · 370989b
2 parents b20d041 + 73fda04
commit 370989b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dags/violation_detection_helpers/utils/prepare_report.py b/dags/violation_detection_helpers/utils/prepare_report.py
@@ -27,9 +27,10 @@ def prepare(self, transformed_documents: list[dict[str, Any]]) -> str | None:
 
             if label and link:
                 # report for single document
-                doc_report = f"Document link: {link} | Label: {label}"
-
-                reports.append(doc_report)
+                # if there was a specific label
+                if label != "None":
+                    doc_report = f"Document link: {link} | Label: {label}"
+                    reports.append(doc_report)
             elif label:
                 logging.error(
                     f"Document with _id: {id} doesn't have "