Fix formatting and return type annotation

dfint · Aug 16, 2024 · bac7da0 · bac7da0
1 parent 50b67b4
commit bac7da0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/search_offsets/search.py b/search_offsets/search.py
@@ -36,7 +36,11 @@ def search(path: str, patterns: list[Pattern]) -> Mapping[str, list[int]]:
     return found
 
 
-def process_found(section_table: SectionTable, pattern_names: Iterable[Pattern], found: Mapping[str, int]) -> dict[str, int]:
+def process_found(
+    section_table: SectionTable,
+    pattern_names: Iterable[Pattern],
+    found: Mapping[str, int],
+) -> Iterable[tuple[str, int]]:
     """
     Print offsets of the found patterns.
     """
@@ -85,7 +89,7 @@ def main(config: DictConfig) -> None:
 
     if config.get("version_name", None):
         result = render_template(processed, checksum=pe.file_header.timedate_stamp, version_name=config.version_name)
-        file_name =  f"offsets_{config.version_name.replace(' ', '_')}.toml"
+        file_name = f"offsets_{config.version_name.replace(' ', '_')}.toml"
         (root_dir / file_name).write_text(result, encoding="utf-8")