From 943030607607cc772aa9b1571974ebdfdbfc1090 Mon Sep 17 00:00:00 2001 From: richardpaulhudson Date: Thu, 22 Dec 2022 15:00:42 +0100 Subject: [PATCH] Handle arrays on GPU --- spacy/pipeline/edit_tree_lemmatizer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py index c65770955..8a26e4aba 100644 --- a/spacy/pipeline/edit_tree_lemmatizer.py +++ b/spacy/pipeline/edit_tree_lemmatizer.py @@ -160,6 +160,8 @@ class EditTreeLemmatizer(TrainablePipe): def _scores2guesses(self, docs, scores): guesses = [] for doc, doc_scores in zip(docs, scores): + if not isinstance(doc_scores, np.ndarray): + doc_scores = doc_scores.get() doc_compat_guesses = [] for i, token in enumerate(doc): for _ in range(self.top_k):