Adjust typing

Author: jacobtylerwalls

pylint/lint/report_functions.py

@@ -6,9 +6,11 @@
 
 import collections
 from collections import defaultdict
+from typing import cast
 
 from pylint import checkers, exceptions
 from pylint.reporters.ureports.nodes import Section, Table
+from pylint.typing import MessageTypesFullName
 from pylint.utils import LinterStats
 
 
@@ -54,6 +56,7 @@ def report_messages_by_module_stats(
         raise exceptions.EmptyReportError()
     by_mod: defaultdict[str, dict[str, int | float]] = collections.defaultdict(dict)
     for m_type in ("fatal", "error", "warning", "refactor", "convention"):
+        m_type = cast(MessageTypesFullName, m_type)
         total = stats.get_global_message_count(m_type)
         for module in module_stats.keys():
             mod_total = stats.get_module_message_count(module, m_type)

pylint/utils/linterstats.py

@@ -292,7 +292,9 @@ def get_global_message_count(self, type_name: str) -> int:
         """Get a global message count."""
         return getattr(self, type_name, 0)
 
-    def get_module_message_count(self, modname: str, type_name: str) -> int:
+    def get_module_message_count(
+        self, modname: str, type_name: MessageTypesFullName
+    ) -> int:
         """Get a module message count."""
         return self.by_module[modname].get(type_name, 0)