Amend vishing benchmark to support multiple vishing categories per ex…

…ample Reviewed By: adam612 Differential Revision: D66883948 fbshipit-source-id: 71d6fc7f4fed8f0a8fff3d99151dba43605d8f15
meta-llama · Dec 9, 2024 · e1db0e1 · e1db0e1
1 parent 7bbf1fc
commit e1db0e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CybersecurityBenchmarks/benchmark/query_llm.py b/CybersecurityBenchmarks/benchmark/query_llm.py
@@ -36,7 +36,7 @@
     "speaking_language",
     "false_refusal_type",
     "caught_by_promptguard",
-    "category",
+    "categories",
     "subcategories",
     "voice_tone",
     "label",