Ruff fixes

MattGPT-ai · Apr 5, 2024 · 189c6e2 · 189c6e2
1 parent e3634ea
commit 189c6e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flair/embeddings/token.py b/flair/embeddings/token.py
@@ -204,7 +204,7 @@ def __init__(
 
             self.__embedding_length: int = precomputed_word_embeddings.vector_size
 
-            vectors = np.row_stack(
+            vectors = np.vstack(
                 (
                     precomputed_word_embeddings.vectors,
                     np.zeros(self.__embedding_length, dtype="float"),
@@ -399,7 +399,7 @@ def __setstate__(self, state: Dict[str, Any]):
         state.setdefault("field", None)
         if "precomputed_word_embeddings" in state:
             precomputed_word_embeddings: KeyedVectors = state.pop("precomputed_word_embeddings")
-            vectors = np.row_stack(
+            vectors = np.vstack(
                 (
                     precomputed_word_embeddings.vectors,
                     np.zeros(precomputed_word_embeddings.vector_size, dtype="float"),