From edc483c379586daf87677ab0bcbfb99514a210d2 Mon Sep 17 00:00:00 2001 From: richardpaulhudson Date: Fri, 9 Dec 2022 14:19:19 +0100 Subject: [PATCH] Improve code readability --- spacy/pipeline/edit_tree_lemmatizer.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py index bbfa622e3..8243dd060 100644 --- a/spacy/pipeline/edit_tree_lemmatizer.py +++ b/spacy/pipeline/edit_tree_lemmatizer.py @@ -165,16 +165,14 @@ class EditTreeLemmatizer(TrainablePipe): eg_d_lowercasing_flags, eg_lowercasing_loss = lowercasing_loss_func( eg_lowercasing_flags, self.model.ops.asarray(lowercasing_truths[i]) ) - d_scores.append( - self.model.ops.xp.hstack( - [ - doc_d_tree_scores, - self.model.ops.asarray( - eg_d_lowercasing_flags, dtype="float32" - ), - ] - ) + doc_d_scores = self.model.ops.xp.hstack( + [ + doc_d_tree_scores, + self.model.ops.asarray(eg_d_lowercasing_flags, dtype="float32"), + ] ) + d_scores.append(doc_d_scores) + loss += eg_lowercasing_loss else: d_scores, loss = tree_loss_func(scores, tree_truths)