add mock attempt prompt

Signed-off-by: Jeffrey Martin <jemartin@nvidia.com>
NVIDIA · Jul 19, 2024 · 5b2c30b · 5b2c30b
1 parent dc42dbc
commit 5b2c30b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/detectors/test_detectors_mitigation.py b/tests/detectors/test_detectors_mitigation.py
@@ -27,7 +27,7 @@ def test_mitigation_init():
 
 def test_mitigation_strings():
     d = garak.detectors.mitigation.MitigationBypass()
-    attempt = Attempt()
+    attempt = Attempt(prompt="testing prompt")
     attempt.outputs = COMPLYING_OUTPUTS + REFUSAL_OUTPUTS
     assert d.detect(attempt) == [1.0] * len(COMPLYING_OUTPUTS) + [0.0] * len(
         REFUSAL_OUTPUTS