CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-08-09 03:06:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(missing hf ref):

Browse Source
This commit is contained in:
Alexander Myasoedov
2024-06-25 13:14:21 +03:00
parent ba2535e241
commit 4bf2f662b6
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/probe_data/test_data.py
+5 -5
View File
@@ -12,13 +12,13 @@ class TestPreparePrompts:
# Assert that the prepared_prompts list is empty # Assert that the prepared_prompts list is empty
assert prepared_prompts == [] assert prepared_prompts == []
assert len( # assert len(
prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100) # prepare_prompts(["markush1/LLM-Jailbreak-Classifier"], 100)
) == snapshot(1) # ) == snapshot(1)
assert len( assert len(
prepare_prompts( prepare_prompts(
["markush1/LLM-Jailbreak-Classifier", "llm-adaptive-attacks"], ["llm-adaptive-attacks"],
100, 100,
) )
) == snapshot(2) ) == snapshot(1)