From 9f1bc3f24c6c9f0412f815abe044274d3840fa23 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 685016b62..c538f02d0 100644 --- a/spacy/training/augment.py +++ b/spacy/training/augment.py @@ -130,7 +130,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]