mypy

mindflowai · Mar 8, 2023 · 34f4c11 · 34f4c11
1 parent 7885b12
commit 34f4c11
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -7,7 +7,7 @@
 def parse_git_diff(diff_str: str):
     diffs = {}
     current_file = None
-    current_diff = []
+    current_diff = []  # type: ignore
 
     excluded_files = []