diff --git a/src/zenml/integrations/llama_index/materializers/document_materializer.py b/src/zenml/integrations/llama_index/materializers/document_materializer.py index 9829232a50..18d4fa558f 100644 --- a/src/zenml/integrations/llama_index/materializers/document_materializer.py +++ b/src/zenml/integrations/llama_index/materializers/document_materializer.py @@ -29,8 +29,12 @@ Document = Any LCDocument = Any else: - from langchain.docstore.document import Document as LCDocument - from llama_index.readers.schema.base import Document + from langchain.docstore.document import ( # type: ignore[import] + Document as LCDocument, + ) + from llama_index.readers.schema.base import ( # type: ignore[import] + Document, + ) class LlamaIndexDocumentMaterializer(LangchainDocumentMaterializer): diff --git a/src/zenml/integrations/llama_index/materializers/gpt_index_materializer.py b/src/zenml/integrations/llama_index/materializers/gpt_index_materializer.py index 51d7070926..c04c368609 100644 --- a/src/zenml/integrations/llama_index/materializers/gpt_index_materializer.py +++ b/src/zenml/integrations/llama_index/materializers/gpt_index_materializer.py @@ -39,8 +39,12 @@ GPTFaissIndex = Any T = TypeVar("T", bound=Any) else: - from llama_index.indices.base import BaseGPTIndex - from llama_index.indices.vector_store import GPTFaissIndex + from llama_index.indices.base import ( # type: ignore[import] + BaseGPTIndex, + ) + from llama_index.indices.vector_store import ( # type: ignore[import] + GPTFaissIndex, + ) T = TypeVar("T", bound=BaseGPTIndex[Any])