Fix augment

This commit is contained in:
Matthew Honnibal 2020-10-05 16:40:23 +02:00
parent 90040aacec
commit 9f1bc3f24c

View File

@ -130,7 +130,7 @@ def make_orth_variants(
for word_idx in range(len(words)): for word_idx in range(len(words)):
for punct_idx in range(len(ndsv)): for punct_idx in range(len(ndsv)):
if ( 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"] and words[word_idx] in ndsv[punct_idx]["variants"]
): ):
words[word_idx] = punct_choices[punct_idx] words[word_idx] = punct_choices[punct_idx]