Spaces:
Sleeping
Sleeping
File size: 751 Bytes
4a1df2e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
"""
Repeat Modification
--------------------------
"""
from textattack.constraints import PreTransformationConstraint
class RepeatModification(PreTransformationConstraint):
"""A constraint disallowing the modification of words which have already
been modified."""
def _get_modifiable_indices(self, current_text):
"""Returns the word indices in current_text which are able to be
deleted."""
try:
return (
set(range(len(current_text.words)))
- current_text.attack_attrs["modified_indices"]
)
except KeyError:
raise KeyError(
"`modified_indices` in attack_attrs required for RepeatModification constraint."
)
|