yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-08 21:28:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

pass faiss batch_size to sqldocumentstore (#2061)

Browse Source
This commit is contained in:
AhmedIdr 2022-01-26 19:35:16 +01:00 committed by GitHub
parent 5079c6847a
commit 488c3e9e52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
haystack/document_stores/faiss.py
View File

@ -272,7 +272,7 @@ class FAISSDocumentStore(SQLDocumentStore):
docs_to_write_in_sql.append(doc)
super(FAISSDocumentStore, self).write_documents(docs_to_write_in_sql, index=index,
duplicate_documents=duplicate_documents)
duplicate_documents=duplicate_documents, batch_size=batch_size)
progress_bar.update(batch_size)
progress_bar.close()
def _create_document_field_map(self) -> Dict: