strict type checking

1 files changed, 6 insertions, 2 deletions

diff --git a/ssg.py b/ssg.py
index da94b03..3b2ee94 100644
--- a/ssg.py
+++ b/ssg.py
@@ -16,7 +16,9 @@ import tomllib
 from jinja2 import Environment, FileSystemLoader
 
 
-def parse_args(p=ArgumentParser(), argv=sys.argv):
+def parse_args(
+    p: ArgumentParser = ArgumentParser(), argv: list[str] = sys.argv
+) -> Namespace:
     p.add_argument(
         "-o", "--output", default=Path("output"), help="Output directory", type=Path
     )
@@ -40,7 +42,9 @@ class Generator:
         self.env = Environment(
             loader=FileSystemLoader("."), autoescape=jinja2.select_autoescape()
         )
-        self.env.globals["prefix"] = args.prefix
+        self.env.globals[  # pyright: ignore [reportUnknownMemberType]
+            "prefix"
+        ] = args.prefix
 
     def write(self, dest: str, content: str | bytes) -> None:
         dest_path: Path = self.output / Path(dest)