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

Merge branch 'INV-temp' of github.com:QData/TextAttack into INV-temp

Browse Source
This commit is contained in:
Jack Morris
2020-09-01 13:14:20 -04:00
parent 8ea872245a be16974b3b
commit f1f14de651
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
textattack/shared/attacked_text.py
View File

@@ -198,6 +198,14 @@ class AttackedText:
raise TypeError(
f"convert_from_original_idxs got invalid idxs type {type(idxs)}"
)
# update length of original_index_map, when the length of AttackedText increased
while len(self.attack_attrs["original_index_map"]) < len(idxs):
missing_index = self.attack_attrs["original_index_map"][-1] + 1
while missing_index in self.attack_attrs["modified_indices"]:
missing_index += 1
self.attack_attrs["original_index_map"] = np.append(
self.attack_attrs["original_index_map"], [missing_index]
)
return [self.attack_attrs["original_index_map"][i] for i in idxs]
def replace_words_at_indices(self, indices, new_words):