From 83624efb92465dc87ba721636ba96849e3d42bde Mon Sep 17 00:00:00 2001 From: voorhs Date: Mon, 25 May 2026 11:09:35 +0300 Subject: [PATCH] tmp fix for faiss typing --- src/autointent/_wrappers/vector_index/faiss.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/autointent/_wrappers/vector_index/faiss.py b/src/autointent/_wrappers/vector_index/faiss.py index 913ee603..08116d2f 100644 --- a/src/autointent/_wrappers/vector_index/faiss.py +++ b/src/autointent/_wrappers/vector_index/faiss.py @@ -5,6 +5,8 @@ from contextlib import contextmanager from typing import TYPE_CHECKING, Any +import numpy as np + from autointent.configs import FaissConfig from autointent.custom_types import Document @@ -61,7 +63,7 @@ def clear_ram(self) -> None: self._index.reset() def get_all_embeddings(self) -> NDArray[Any]: - return self._index.reconstruct_n(0, self._index.ntotal) # type: ignore[no-any-return] + return np.asarray(self._index.reconstruct_n(0, self._index.ntotal)) def add(self, embeddings: NDArray[Any], documents: list[Document]) -> None: self._index.add(embeddings)