From dd5975ad127aa47770b0f8265994f5f7788ff701 Mon Sep 17 00:00:00 2001 From: Richard Hudson Date: Tue, 17 Jan 2023 15:11:20 +0100 Subject: [PATCH] Improve naming MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Daniël de Kok --- spacy/pipeline/edit_tree_lemmatizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py index ad4f266db..0c21bc1bf 100644 --- a/spacy/pipeline/edit_tree_lemmatizer.py +++ b/spacy/pipeline/edit_tree_lemmatizer.py @@ -198,7 +198,7 @@ class EditTreeLemmatizer(TrainablePipe): doc_scores = self.numpy_ops.asarray(doc_scores) doc_compat_guesses = [] for i, token in enumerate(doc): - for _ in range(predictions_to_consider): + for _ in range(top_k): candidate = int(doc_scores[i].argmax()) candidate_tree_id = self.cfg["labels"][candidate] if self.trees.apply(candidate_tree_id, token.text) is not None: