Improve typing

allenporter · Aug 3, 2024 · f1ada62 · f1ada62
1 parent 40f4715
commit f1ada62
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/home_assistant_datasets/tool/assist/conftest.py b/home_assistant_datasets/tool/assist/conftest.py
@@ -74,7 +74,7 @@ def pytest_generate_tests(metafunc: Any) -> None:
 
 
 @pytest.fixture(autouse=True)
-def restore_tz() -> Generator[None, None]:
+def restore_tz() -> Generator[None, None, None]:
     yield
     # Home Assistant teardown seems to run too soon and expects this so try to
     # patch it in first.

diff --git a/script/export.py b/script/export.py
@@ -5,7 +5,7 @@
 import sys
 
 
-def main():
+def main() -> None:
     lines = sys.stdin.readlines()
     for line in lines:
         record = json.loads(line)

diff --git a/script/human_eval_metrics.py b/script/human_eval_metrics.py
@@ -94,7 +94,7 @@ def main() -> None:
         print(yaml.dump(model_results, sort_keys=True, explicit_start=True))
 
         if args.samples:
-            limited_samples = {}
+            limited_samples: dict[str, dict[str, str]] = {}
             for model_id in model_samples:
                 limited_samples[model_id] = {}
                 for label in model_samples[model_id]: