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

speed up and fixes

This commit is contained in:
uvafan
2020-05-17 20:33:01 -04:00
parent b94684971e
commit edf01eb4c8
8 changed files with 9 additions and 9 deletions

View File

@@ -76,7 +76,7 @@ class Attack:
"""
if not self.transformation:
raise RuntimeError('Cannot call `get_transformations` without a transformation.')
transformations = np.array(self.transformation(text,
modification_constraints=self.modification_constraints, **kwargs))
if apply_constraints:
@@ -138,7 +138,7 @@ class Attack:
if final_result.succeeded:
return SuccessfulAttackResult(initial_result, final_result)
else:
return FailedAttackResult(intial_result, final_result)
return FailedAttackResult(initial_result, final_result)
def _get_examples_from_dataset(self, dataset, num_examples=None, shuffle=False,
attack_n=False, attack_skippable_examples=False):