From 4ed3e037df766aa2f2827a4b1a63a1f80a79485b Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 5 Oct 2020 16:40:23 +0200 Subject: [PATCH] Fix augment --- spacy/training/augment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/training/augment.py b/spacy/training/augment.py index 06656bdd8..7db8919e9 100644 --- a/spacy/training/augment.py +++ b/spacy/training/augment.py @@ -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]