fix(fmt):

This commit is contained in:
Alexander Myasoedov
2024-11-30 00:02:31 +02:00
parent 107181fae2
commit f57f3e9f43
21 changed files with 41 additions and 116 deletions
-3
View File
@@ -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 = []