remove trailing whitespace, sort imports, remove unused arguments

This commit is contained in:
asoderlind 2023-10-16 11:22:42 +02:00
parent e2a8ca143a
commit e73636bef3

View File

@ -1,5 +1,5 @@
from application.vectorstore.base import BaseVectorStore
from langchain.vectorstores import FAISS from langchain.vectorstores import FAISS
from application.vectorstore.base import BaseVectorStore
from application.core.settings import settings from application.core.settings import settings
class FaissStore(BaseVectorStore): class FaissStore(BaseVectorStore):
@ -27,7 +27,7 @@ class FaissStore(BaseVectorStore):
def save_local(self, *args, **kwargs): def save_local(self, *args, **kwargs):
return self.docsearch.save_local(*args, **kwargs) return self.docsearch.save_local(*args, **kwargs)
def assert_embedding_dimensions(self, embeddings, *args, **kwargs): def assert_embedding_dimensions(self, embeddings):
""" """
Check that the word embedding dimension of the docsearch index matches Check that the word embedding dimension of the docsearch index matches
the dimension of the word embeddings used the dimension of the word embeddings used