From 9ccd38360ad993497d309b705b661c8b511f983a Mon Sep 17 00:00:00 2001 From: noman-code-droid Date: Tue, 23 Jun 2026 22:43:02 +0500 Subject: [PATCH] fix: add type annotations to download_tiles.py --- train/sft/download_tiles.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train/sft/download_tiles.py b/train/sft/download_tiles.py index d27761e..9917826 100644 --- a/train/sft/download_tiles.py +++ b/train/sft/download_tiles.py @@ -34,7 +34,7 @@ def shard_suffix(p: str) -> str: return p -def collect_paths(retrieval_dir: Path, splits: list[str]) -> set[str]: +def collect_paths(retrieval_dir: Path, splits: list[str]) -> dict[str, str | None]: """Collect every unique absolute path across hit lists + gold suffixes. Gold paths use the dataset-relative form ('images/shard_.../chunk.png'); @@ -111,7 +111,7 @@ def fetch_tile( return False, f"{last_err}" -def main(): +def main() -> None: p = argparse.ArgumentParser() p.add_argument( "--retrieval-dir",