Add missing type hints

Co-authored-by: Will Langdale <[email protected]>
uktrade · Jan 6, 2025 · 09a4b63 · 09a4b63
1 parent f4eaf79
commit 09a4b63
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/matchbox/common/factories.py b/src/matchbox/common/factories.py
@@ -6,7 +6,7 @@
 import rustworkx as rx
 
 
-def verify_components(table) -> dict:
+def verify_components(table: pa.Table) -> dict:
     """
     Fast verification of connected components using rustworkx.
 

diff --git a/src/matchbox/server/postgresql/benchmark/generate_tables.py b/src/matchbox/server/postgresql/benchmark/generate_tables.py
@@ -51,7 +51,7 @@ def _hash_list_int(li: list[int]) -> list[bytes]:
     return [HASH_FUNC(str(i).encode("utf-8")).digest() for i in li]
 
 
-def _unique_clusters(all_parents: Iterable[int], all_probabilities: Iterable[int]):
+def _unique_clusters(all_parents: Iterable[int], all_probabilities: Iterable[int]) -> tuple[list[int], list[float]]:
     ll = set()
     clusters = []
     probabilities = []