fix(missing hf ref):

msoedov · Jun 25, 2024 · 4bf2f66 · 4bf2f66
1 parent ba2535e
commit 4bf2f66
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/agentic_security/probe_data/test_data.py b/agentic_security/probe_data/test_data.py
@@ -12,13 +12,13 @@ def test_empty_dataset_list(self):
         # Assert that the prepared_prompts list is empty
         assert prepared_prompts == []
 
-        assert len(
-            prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100)
-        ) == snapshot(1)
+        # assert len(
+        #     prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100)
+        # ) == snapshot(1)
 
         assert len(
             prepare_prompts(
-                ["markush1/LLM-Jailbreak-Classifier", "llm-adaptive-attacks"],
+                ["llm-adaptive-attacks"],
                 100,
             )
-        ) == snapshot(2)
+        ) == snapshot(1)