--- a/app.py
+++ b/app.py
@@ -1,2 +1,2 @@
-def inventory(sku, records):
+def inventory(sku: str, records: dict[str, int]) -> int | None:
     return records.get(sku)