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

change args to reference_text

Browse Source
This commit is contained in:
Jin Yong Yoo
2020-06-30 10:48:10 -04:00
parent 87ec3ad1bc
commit 769f160fa0
19 changed files with 97 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
textattack/shared/attack.py
View File

@@ -122,9 +122,15 @@ class Attack:
for C in self.constraints:
if len(filtered_texts) == 0:
break
filtered_texts = C.call_many(
filtered_texts, current_text, original_text=original_text
)
if C.compare_against_original:
if not original_text:
raise ValueError(
f"Missing `original_text` argument when constraint {type(C)} is set to compare against `original_text`"
)
filtered_texts = C.call_many(filtered_texts, original_text)
else:
filtered_texts = C.call_many(filtered_texts, current_text)
# Default to false for all original transformations.
for original_transformed_text in transformed_texts:
self.constraints_cache[(current_text, original_transformed_text)] = False