From 361b64e64880295ea78b175580785e7497d4944e Mon Sep 17 00:00:00 2001
From: Richard Hudson <richard@explosion.ai>
Date: Fri, 23 Dec 2022 14:37:46 +0100
Subject: [PATCH] Use NumpyOps() to copy data to CPU

Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com>
---
 spacy/pipeline/edit_tree_lemmatizer.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/spacy/pipeline/edit_tree_lemmatizer.py b/spacy/pipeline/edit_tree_lemmatizer.py
index 45ed59420..9c5ca0846 100644
--- a/spacy/pipeline/edit_tree_lemmatizer.py
+++ b/spacy/pipeline/edit_tree_lemmatizer.py
@@ -160,8 +160,7 @@ 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()
+            NumpyOps().asarray(doc_scores)
             doc_compat_guesses = []
             for i, token in enumerate(doc):
                 for _ in range(self.top_k):