Fix augment

This commit is contained in:
Matthew Honnibal 2020-10-05 16:40:23 +02:00
parent 6a9d14e35a
commit 4ed3e037df

View File

@ -131,7 +131,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]