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

add LM constraint to args + formatting

Browse Source
This commit is contained in:
Jack Morris
2020-06-28 11:32:00 -04:00
parent 5f3e2b2961
commit f588200119
10 changed files with 166 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/constraints/grammaticality/language_models/language_model_constraint.py
View File

@@ -31,7 +31,7 @@ class LanguageModelConstraint(ABC, Constraint):
def _check_constraint(self, transformed_text, current_text, original_text=None):
if self.compare_against_original:
current_text = original_text
try:
indices = transformed_text.attack_attrs["newly_modified_indices"]
except KeyError: