style: Finish type-hinting

Author: jmgate

example/ex_6_creating_retryable_stages.py

@@ -27,7 +27,7 @@ def __init__(
         console_force_terminal: Optional[bool] = None,
         console_log_path: bool = True,
         print_commands: bool = True,
-    ):
+    ) -> None:
         super().__init__(
             stages,
             console_force_terminal=console_force_terminal,

example/ex_7_customizing_the_summary.py

@@ -29,7 +29,7 @@ def __init__(
         console_force_terminal: Optional[bool] = None,
         console_log_path: bool = True,
         print_commands: bool = True,
-    ):
+    ) -> None:
         super().__init__(
             stages,
             console_force_terminal=console_force_terminal,

pyproject.toml

@@ -74,7 +74,7 @@ line-length = 79
 extend-select = [
     "A",
     "AIR",
-    # "ANN",
+    "ANN",
     "ARG",
     "ASYNC",
     "B",

staged_script/staged_script.py

@@ -26,7 +26,7 @@
 from datetime import datetime, timedelta, timezone
 from pathlib import Path
 from subprocess import CompletedProcess
-from typing import Callable, NamedTuple, Optional
+from typing import Any, Callable, NamedTuple, NoReturn, Optional
 
 import __main__
 import rich.traceback
@@ -145,7 +145,7 @@ def __init__(
         console_force_terminal: Optional[bool] = None,
         console_log_path: bool = True,
         print_commands: bool = True,
-    ):
+    ) -> None:
         """
         Initialize a :class:`StagedScript` object.
 
@@ -248,7 +248,7 @@ def stage(stage_name: str, heading: str) -> Callable:
 
         def decorator(func: Callable) -> Callable:
             def get_phase_method(  # noqa: D417
-                self,
+                self,  # noqa: ANN001
                 method_name: str,
             ) -> Callable:
                 """
@@ -273,9 +273,9 @@ def get_phase_method(  # noqa: D417
                 )
 
             def run_retryable_phases(  # noqa: D417
-                self,
-                *args,
-                **kwargs,
+                self,  # noqa: ANN001
+                *args: Any,  # noqa: ANN401
+                **kwargs: Any,  # noqa: ANN401
             ) -> None:
                 """
                 Run the retryable phases.
@@ -310,7 +310,7 @@ def run_retryable_phases(  # noqa: D417
                 get_phase_method(self, "_end_stage")()
 
             @functools.wraps(func)
-            def wrapper(self, *args, **kwargs) -> None:
+            def wrapper(self, *args: Any, **kwargs: Any) -> None:  # noqa: ANN001, ANN401
                 """
                 Turn a function into a stage.
 
@@ -759,7 +759,7 @@ def parse_args(self, argv: list[str]) -> None:
             ]:
                 setattr(self, retry_arg, getattr(self.args, retry_arg, None))
 
-    def raise_parser_error(self, message):
+    def raise_parser_error(self, message: str) -> NoReturn:
         """
         Raise a parser error.
 
@@ -788,7 +788,7 @@ def run(
         *,
         pretty_print: bool = False,
         print_command: Optional[bool] = None,
-        **kwargs,
+        **kwargs: Any,  # noqa: ANN401
     ) -> CompletedProcess:
         """
         Run a command in the underlying shell.