Wrap opensearch imports into safe_import (#2907)

* Wrap opensearch imports into `safe_import`

* black
This commit is contained in:
Sara Zan 2022-07-28 12:25:31 +02:00 committed by GitHub
parent e7627c3f8b
commit 330a1c0249
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -12,8 +12,11 @@ from haystack.document_stores.es_converter import (
elasticsearch_index_to_document_store,
open_search_index_to_document_store,
)
from haystack.document_stores.opensearch import OpenSearchDocumentStore, OpenDistroElasticsearchDocumentStore
OpenSearchDocumentStore = safe_import("haystack.document_stores.opensearch", "OpenSearchDocumentStore", "opensearch")
OpenDistroElasticsearchDocumentStore = safe_import(
"haystack.document_stores.opensearch", "OpenDistroElasticsearchDocumentStore", "opensearch"
)
SQLDocumentStore = safe_import("haystack.document_stores.sql", "SQLDocumentStore", "sql")
FAISSDocumentStore = safe_import("haystack.document_stores.faiss", "FAISSDocumentStore", "faiss")
PineconeDocumentStore = safe_import("haystack.document_stores.pinecone", "PineconeDocumentStore", "pinecone")