diff --git a/dags/violation_detection_helpers/utils/classify.py b/dags/violation_detection_helpers/utils/classify.py index a6040c65..232d8244 100644 --- a/dags/violation_detection_helpers/utils/classify.py +++ b/dags/violation_detection_helpers/utils/classify.py @@ -42,6 +42,7 @@ def classify(self, text: str) -> str: + "\n\n[VIOLATION DETECTIONS]\n", }, ], + n=1, ) # the label to add to general data structure