diff --git a/garak/buffs/lowercase.py b/garak/buffs/lowercase.py index a84780c2..8d9f621b 100644 --- a/garak/buffs/lowercase.py +++ b/garak/buffs/lowercase.py @@ -6,7 +6,6 @@ """ Buff that converts prompts to lower case. """ from collections.abc import Iterable -from typing import List import garak.attempt from garak.buffs.base import Buff diff --git a/garak/buffs/paraphrase.py b/garak/buffs/paraphrase.py index 04bf50c9..e6f06b48 100644 --- a/garak/buffs/paraphrase.py +++ b/garak/buffs/paraphrase.py @@ -6,7 +6,6 @@ """ Buff that paraphrases a prompt. """ from collections.abc import Iterable -import copy import torch from transformers import AutoTokenizer, AutoModelForSeq2SeqLM diff --git a/tests/test_config.py b/tests/test_config.py index 293313dc..d0513cd8 100644 --- a/tests/test_config.py +++ b/tests/test_config.py @@ -439,3 +439,13 @@ def remove_laurelhurst_log(): os.remove("laurelhurst.report.jsonl") request.addfinalizer(remove_laurelhurst_log) + + +def test_buff_config_assertion(): + importlib.reload(_config) + import garak._plugins + + test_value = 9001 + _config.plugins.buffs["paraphrase.Fast"] = {"num_beams": test_value} + p = garak._plugins.load_plugin("buffs.paraphrase.Fast") + assert p.num_beams == test_value