Skip to content

Commit

Permalink
[chore] fix lint issues (mem0ai#607)
Browse files Browse the repository at this point in the history
  • Loading branch information
deshraj committed Sep 13, 2023
1 parent 0f23d5f commit 701d0b2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
2 changes: 1 addition & 1 deletion embedchain/llm/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ def query(self, input_query: str, contexts: List[str], config: BaseLlmConfig = N
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):
"""
Expand Down
5 changes: 1 addition & 4 deletions embedchain/loaders/mdx.py
Original file line number Diff line number Diff line change
@@ -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,
Expand Down

0 comments on commit 701d0b2

Please sign in to comment.