From 701d0b21ef5a0516004e6cf4cc81806cb92f2188 Mon Sep 17 00:00:00 2001 From: Deshraj Yadav Date: Tue, 12 Sep 2023 20:04:31 -0700 Subject: [PATCH] [chore] fix lint issues (#607) --- embedchain/llm/base.py | 2 +- embedchain/loaders/mdx.py | 5 +---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/embedchain/llm/base.py b/embedchain/llm/base.py index cc0179ee..6c24ba28 100644 --- a/embedchain/llm/base.py +++ b/embedchain/llm/base.py @@ -204,7 +204,7 @@ class BaseLlm(JSONSerializable): finally: if config: # Restore previous config - self.config: BaseLlmConfig = BaseLlmConfig.deserialize(prev_config) + self.config: BaseLlmConfig = BaseLlmConfig.deserialize(prev_config) def chat(self, input_query: str, contexts: List[str], config: BaseLlmConfig = None, dry_run=False): """ diff --git a/embedchain/loaders/mdx.py b/embedchain/loaders/mdx.py index df1bd26c..9d73b6ca 100644 --- a/embedchain/loaders/mdx.py +++ b/embedchain/loaders/mdx.py @@ -1,17 +1,14 @@ import hashlib -from langchain.document_loaders import PyPDFLoader - from embedchain.helper.json_serializable import register_deserializable from embedchain.loaders.base_loader import BaseLoader -from embedchain.utils import clean_string @register_deserializable class MdxLoader(BaseLoader): def load_data(self, url): """Load data from a mdx file.""" - with open(url, 'r', encoding="utf-8") as infile: + with open(url, "r", encoding="utf-8") as infile: content = infile.read() meta_data = { "url": url,