mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-07-09 21:18:36 +02:00
fix(fmt):
This commit is contained in:
@@ -152,7 +152,6 @@ def load_dataset_v6():
|
||||
|
||||
@cache_to_disk()
|
||||
def load_dataset_v7():
|
||||
|
||||
splits = {
|
||||
"mini_JailBreakV_28K": "JailBreakV_28K/mini_JailBreakV_28K.csv",
|
||||
"JailBreakV_28K": "JailBreakV_28K/JailBreakV_28K.csv",
|
||||
@@ -173,7 +172,6 @@ def load_dataset_v7():
|
||||
|
||||
@cache_to_disk()
|
||||
def load_dataset_v8():
|
||||
|
||||
df = pd.read_csv(
|
||||
"hf://datasets/ShawnMenz/jailbreak_sft_rm_ds/jailbreak_sft_rm_ds.csv",
|
||||
names=["jailbreak", "prompt"],
|
||||
@@ -321,7 +319,6 @@ class Stenography:
|
||||
|
||||
def apply(self):
|
||||
for prompt_group in self.prompt_groups:
|
||||
|
||||
size = len(prompt_group.prompts)
|
||||
for name, fn in self.fn_library.items():
|
||||
logger.info(f"Applying {name} to {prompt_group.dataset_name}")
|
||||
|
||||
@@ -9,7 +9,6 @@ url = "https://raw.githubusercontent.com/tml-epfl/llm-adaptive-attacks/main/harm
|
||||
|
||||
class Module:
|
||||
def __init__(self, prompt_groups: []):
|
||||
|
||||
r = httpx.get(url)
|
||||
|
||||
content = r.content
|
||||
|
||||
@@ -4,7 +4,6 @@ from .adaptive_attacks import Module
|
||||
|
||||
|
||||
class TestModule:
|
||||
|
||||
# Module can be initialized with a list of prompt groups.
|
||||
def test_initialize_with_prompt_groups(self):
|
||||
prompt_groups = []
|
||||
|
||||
Reference in New Issue
Block a user