Add mypy (#224)

.gitignore

@@ -30,6 +30,7 @@ ENV/
 .pytest_cache/
 .ruff_cache/
 .coverage
+.mypy_cache/
 
 # macOS specific files
 .DS_Store

.pre-commit-config.yaml

@@ -25,3 +25,8 @@ repos:
        id: detect-private-key
        id: end-of-file-fixer
        id: trailing-whitespace
+
+-   repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.3.0
+    hooks:
+    -   id: mypy

requirements.txt

@@ -3,4 +3,5 @@ openai==0.27.8
 pre-commit==3.3.3
 ruff==0.0.272
 typer==0.9.0
-pytest==7.3.1
+pytest==7.3.1
+mypy==1.3.0

scripts/benchmark.py

@@ -6,6 +6,7 @@ import subprocess
 
 from itertools import islice
 from pathlib import Path
+from typing import Iterable
 
 from typer import run
 
@@ -15,7 +16,7 @@ def main(
 ):
     path = Path("benchmark")
 
-    folders = path.iterdir()
+    folders: Iterable[Path] = path.iterdir()
 
     if n_benchmarks:
         folders = islice(folders, n_benchmarks)