From 5240baccfee33af84c3b92813da827f4c5bbd7fa Mon Sep 17 00:00:00 2001
From: kadarakos <kadar.akos@gmail.com>
Date: Mon, 4 Jul 2022 17:15:33 +0200
Subject: [PATCH] dont use get_array_module (#11056)

---
 spacy/pipeline/textcat.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spacy/pipeline/textcat.py b/spacy/pipeline/textcat.py
index bc3f127fc..c45f819fc 100644
--- a/spacy/pipeline/textcat.py
+++ b/spacy/pipeline/textcat.py
@@ -192,7 +192,7 @@ class TextCategorizer(TrainablePipe):
         if not any(len(doc) for doc in docs):
             # Handle cases where there are no tokens in any docs.
             tensors = [doc.tensor for doc in docs]
-            xp = get_array_module(tensors)
+            xp = self.model.ops.xp
             scores = xp.zeros((len(list(docs)), len(self.labels)))
             return scores
         scores = self.model.predict(docs)