diff --git a/src/formatters/formatters.py b/src/formatters/formatters.py index 0820fbc..3e04407 100644 --- a/src/formatters/formatters.py +++ b/src/formatters/formatters.py @@ -53,7 +53,7 @@ class TerminalFormatter(BaseFormatter): return text def format(self, result: ReviewResult) -> str: - output = [] + output: list[Panel | Table] = [] if result.error: output.append(Panel( @@ -132,10 +132,10 @@ class MarkdownFormatter(BaseFormatter): def get_formatter(format_type: str = "terminal", **kwargs) -> BaseFormatter: - formatters = { + formatters: dict[str, type[BaseFormatter]] = { "terminal": TerminalFormatter, "json": JSONFormatter, "markdown": MarkdownFormatter, } formatter_class = formatters.get(format_type, TerminalFormatter) - return formatter_class(**kwargs) + return formatter_class(**kwargs) # type: ignore[arg-type]