1
0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00

update black formatting

This commit is contained in:
Jack Morris
2020-08-30 15:25:56 -04:00
parent b014a75f9d
commit af7f31d265
13 changed files with 100 additions and 35 deletions

View File

@@ -68,7 +68,8 @@ class Attack:
self.pre_transformation_constraints = []
for constraint in constraints:
if isinstance(
constraint, textattack.constraints.PreTransformationConstraint,
constraint,
textattack.constraints.PreTransformationConstraint,
):
self.pre_transformation_constraints.append(constraint)
else:
@@ -95,8 +96,10 @@ class Attack:
self.search_method.get_transformations = self.get_transformations
# The search method only needs access to the first argument. The second is only used
# by the attack class when checking whether to skip the sample
self.search_method.get_goal_results = lambda attacked_text_list: self.goal_function.get_results(
attacked_text_list
self.search_method.get_goal_results = (
lambda attacked_text_list: self.goal_function.get_results(
attacked_text_list
)
)
self.search_method.filter_transformations = self.filter_transformations
@@ -248,11 +251,20 @@ class Attack:
final_result = self.search_method(initial_result)
self.clear_cache()
if final_result.goal_status == GoalFunctionResultStatus.SUCCEEDED:
return SuccessfulAttackResult(initial_result, final_result,)
return SuccessfulAttackResult(
initial_result,
final_result,
)
elif final_result.goal_status == GoalFunctionResultStatus.SEARCHING:
return FailedAttackResult(initial_result, final_result,)
return FailedAttackResult(
initial_result,
final_result,
)
elif final_result.goal_status == GoalFunctionResultStatus.MAXIMIZING:
return MaximizedAttackResult(initial_result, final_result,)
return MaximizedAttackResult(
initial_result,
final_result,
)
else:
raise ValueError(f"Unrecognized goal status {final_result.goal_status}")