Skip to content

Commit e26492a

Browse files
authored
Merge pull request #252 from rogier-stegeman/fix-empty-list
fix(augmenter/word/random.py): Empty list error
2 parents 30b2b0b + 879225a commit e26492a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nlpaug/augmenter/word/random.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def __init__(self, action=Action.DELETE, name='RandomWord_Aug', aug_min=1, aug_m
3636
tokenizer=tokenizer, reverse_tokenizer=reverse_tokenizer, device='cpu', verbose=verbose,
3737
stopwords_regex=stopwords_regex, include_detail=False)
3838

39-
self.target_words = ['_'] if target_words is None else target_words
39+
self.target_words = target_words or ['_']
4040

4141
# https://arxiv.org/pdf/1711.02173.pdf, https://arxiv.org/pdf/1809.02079.pdf, https://arxiv.org/pdf/1903.09460.pdf
4242
def swap(self, data):

0 commit comments

Comments
 (0)