Use NumpyOps() to copy data to CPU

Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com>
This commit is contained in:
Richard Hudson 2022-12-23 14:37:46 +01:00 committed by GitHub
parent 95a4835342
commit 361b64e648
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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):