From 4bf2f662b6028d9633b886b2a238c6d0caafc485 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Tue, 25 Jun 2024 13:14:21 +0300 Subject: [PATCH] fix(missing hf ref): --- agentic_security/probe_data/test_data.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/agentic_security/probe_data/test_data.py b/agentic_security/probe_data/test_data.py index 6e391c8..a4398f6 100644 --- a/agentic_security/probe_data/test_data.py +++ b/agentic_security/probe_data/test_data.py @@ -12,13 +12,13 @@ class TestPreparePrompts: # 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)