Adapt widgets and test to corpus refactoring

biolab · Jan 12, 2022 · d60fbcb · d60fbcb
1 parent 9207553
commit d60fbcb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/orangecontrib/text/widgets/owscoredocuments.py b/orangecontrib/text/widgets/owscoredocuments.py
@@ -142,8 +142,9 @@ def _preprocess_words(
     #  cases like this think about implementation of preprocessors for a list
     #  of strings
     words_feature = StringVariable("words")
-    words_c = Corpus(
+    words_c = Corpus.from_numpy(
         Domain([], metas=[words_feature]),
+        np.empty((len(words), 0)),
         metas=np.array([[w] for w in words]),
         text_features=[words_feature],
     )
@@ -570,7 +571,7 @@ def _send_output(self) -> None:
                 metas=d.metas + tuple(ContinuousVariable(get_unique_names(d, l))
                                       for l in labels),
             )
-            out_corpus = Corpus(
+            out_corpus = Corpus.from_numpy(
                 domain,
                 self.corpus.X,
                 self.corpus.Y,

diff --git a/orangecontrib/text/widgets/tests/test_owscoredocuments.py b/orangecontrib/text/widgets/tests/test_owscoredocuments.py
@@ -26,7 +26,7 @@ def embedding_mock(_, corpus, __):
         return np.ones((len(corpus), 10))
     else:  # corpus is Corpus
         return (
-            Corpus(
+            Corpus.from_numpy(
                 domain=Domain([ContinuousVariable(str(i)) for i in range(10)]),
                 X=np.ones((len(corpus), 10)),
             ),
@@ -168,8 +168,9 @@ def create_corpus(texts: List[str]) -> Corpus:
         """Create sample corpus with texts passed"""
         text_var = StringVariable("Text")
         domain = Domain([], metas=[text_var])
-        c = Corpus(
+        c = Corpus.from_numpy(
             domain,
+            X=np.empty((len(texts), 0)),
             metas=np.array(texts).reshape(-1, 1),
             text_features=[text_var],
         )