From fc279a9c7cdd85b8f2032718fcc3c9f3874b3653 Mon Sep 17 00:00:00 2001 From: Logan Markewich Date: Fri, 17 Nov 2023 15:56:17 -0600 Subject: [PATCH] linting --- llama_hub/obsidian/base.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/llama_hub/obsidian/base.py b/llama_hub/obsidian/base.py index 659c0d5336..0417803b68 100644 --- a/llama_hub/obsidian/base.py +++ b/llama_hub/obsidian/base.py @@ -7,9 +7,11 @@ """ import os from pathlib import Path -from typing import Any, List +from typing import Any, List, TYPE_CHECKING + +if TYPE_CHECKING: + from langchain.docstore.document import Document as LCDocument -from langchain.docstore.document import Document as LCDocument from llama_index.readers.base import BaseReader from llama_index.readers.file.markdown_reader import MarkdownReader from llama_index.readers.schema.base import Document @@ -39,7 +41,7 @@ def load_data(self, *args: Any, **load_kwargs: Any) -> List[Document]: docs.extend(content) return docs - def load_langchain_documents(self, **load_kwargs: Any) -> List[LCDocument]: + def load_langchain_documents(self, **load_kwargs: Any) -> List["LCDocument"]: """Load data in LangChain document format.""" docs = self.load_data(**load_kwargs) return [d.to_langchain_format() for d in docs]