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 in attack-results-take-goal-functions

Browse Source
This commit is contained in:
Jack Morris
2020-05-01 21:40:43 -04:00
parent 6adbb55b4d a8eeb7361c
commit b26ed34224
140 changed files with 6077 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
textattack/shared/utils.py
View File

@@ -109,13 +109,6 @@ def default_class_repr(self):
extra_str = ''
extra_str = extra_str.format(**self.__dict__)
return f'{self.__class__.__name__}{extra_str}'
def color_text(label, color=None, method=None):
if method == 'file':
method = None
if color is None:
color = color_from_label(label)
return color_text_by_method(str(label), color, method)
LABEL_COLORS = [
'red', 'green',
@@ -130,7 +123,7 @@ def color_from_label(label_num):
label_num %= len(LABEL_COLORS)
return LABEL_COLORS[label_num]
def color_text_by_method(text, color=None, method=None):
def color_text(text, color=None, method=None):
if method is None:
return text
if method == 'html':