Fix augment

This commit is contained in:
Matthew Honnibal 2020-10-05 16:40:23 +02:00
parent 6a9d14e35a
commit 4ed3e037df
1 changed files with 1 additions and 1 deletions

View File

@ -131,7 +131,7 @@ def make_orth_variants(
for word_idx in range(len(words)):
for punct_idx in range(len(ndsv)):
if (
tags[word_idx] in ndsv[punct_idx]["TAG"]
tags[word_idx] in ndsv[punct_idx]["tags"]
and words[word_idx] in ndsv[punct_idx]["variants"]
):
words[word_idx] = punct_choices[punct_idx]