Merge pull request #3606 from flairNLP/scipy_update

fix: update scipy .A to toarray()
flairNLP · Feb 1, 2025 · 005ec45 · 005ec45
2 parents e2865f7 + 0b95bcd
commit 005ec45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flair/embeddings/document.py b/flair/embeddings/document.py
@@ -210,7 +210,7 @@ def embed(self, sentences: Union[list[Sentence], Sentence]):
             sentences = [sentences]
 
         raw_sentences = [s.to_original_text() for s in sentences]
-        tfidf_vectors = torch.from_numpy(self.vectorizer.transform(raw_sentences).A)
+        tfidf_vectors = torch.from_numpy(self.vectorizer.transform(raw_sentences).toarray())
 
         for sentence_id, sentence in enumerate(sentences):
             sentence.set_embedding(self.name, tfidf_vectors[sentence_id])