fix bugs for LabelConstraint

microsoft · Aug 20, 2023 · 3c3c8d9 · 3c3c8d9
1 parent d5252bf
commit 3c3c8d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prompt_attack/label_constraint.py b/prompt_attack/label_constraint.py
@@ -9,7 +9,7 @@ class LabelConstraint(PreTransformationConstraint):
     """
 
     def __init__(self, labels=[]):
-        self.labels = labels
+        self.labels = [label.lower() for label in labels]
 
     def _get_modifiable_indices(self, current_text):
         modifiable_indices = set()