alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

attack recipes -> classes

Browse Source
This commit is contained in:
Jack Morris
2020-08-06 07:42:47 -04:00
parent adbda1ea3b
commit ae575cfeb5
21 changed files with 781 additions and 677 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
textattack/commands/attack/attack_args_helpers.py
View File

@@ -194,9 +194,9 @@ def parse_attack_from_args(args):
recipe_name, params = args.recipe.split(ARGS_SPLIT_TOKEN)
if recipe_name not in ATTACK_RECIPE_NAMES:
raise ValueError(f"Error: unsupported recipe {recipe_name}")
recipe = eval(f"{ATTACK_RECIPE_NAMES[recipe_name]}(model, {params})")
recipe = eval(f"{ATTACK_RECIPE_NAMES[recipe_name]}.build(model, {params})")
elif args.recipe in ATTACK_RECIPE_NAMES:
recipe = eval(f"{ATTACK_RECIPE_NAMES[args.recipe]}(model)")
recipe = eval(f"{ATTACK_RECIPE_NAMES[args.recipe]}.build(model)")
else:
raise ValueError(f"Invalid recipe {args.recipe}")
recipe.goal_function.query_budget = args.query_budget