From 34f4c11a7ec6954f111a080a2b5ef9d03c2797e2 Mon Sep 17 00:00:00 2001 From: nollied Date: Wed, 8 Mar 2023 00:49:37 +0000 Subject: [PATCH] mypy --- mindflow/core/git/diff.py | 2 +- mindflow/utils/diff_parser.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindflow/core/git/diff.py b/mindflow/core/git/diff.py index e71dfc2..eddaa41 100644 --- a/mindflow/core/git/diff.py +++ b/mindflow/core/git/diff.py @@ -74,7 +74,7 @@ def run_diff(args: Tuple[str]) -> str: def batch_git_diffs( - file_diffs: List[Dict[str, str]], token_limit: int + file_diffs: Dict[str, str], token_limit: int ) -> List[List[Tuple[str, str]]]: batches = [] current_batch: List = [] diff --git a/mindflow/utils/diff_parser.py b/mindflow/utils/diff_parser.py index 44739f6..03636e8 100644 --- a/mindflow/utils/diff_parser.py +++ b/mindflow/utils/diff_parser.py @@ -7,7 +7,7 @@ def parse_git_diff(diff_str: str): diffs = {} current_file = None - current_diff = [] + current_diff = [] # type: ignore excluded_files = []