chore: fix mypy issues

Author: lepture

src/mistune/core.py

@@ -17,6 +17,7 @@
     Type,
     TypeVar,
     Union,
+    cast,
 )
 
 if sys.version_info >= (3, 11):
@@ -246,7 +247,7 @@ def render_wiki(renderer, key, title):
 
     def _get_method(self, name: str) -> Callable[..., str]:
         try:
-            return object.__getattribute__(self, name)
+            return cast(Callable[..., str], object.__getattribute__(self, name))
         except AttributeError:
             method = self.__methods.get(name)
             if not method:

src/mistune/inline_parser.py

@@ -9,6 +9,7 @@
     Optional,
     Set,
     Tuple,
+    cast,
 )
 
 from .core import InlineState, Parser
@@ -717,7 +718,7 @@ def _has_emphasis_enabled(parts: List[Dict[str, Any]], opener: _Delimiter, close
 
 def _text_raw(token: Dict[str, Any]) -> str:
     if token["type"] == "text":
-        return token["raw"]
+        return cast(str, token["raw"])
     return ""
 
 

src/mistune/toc.py

@@ -1,5 +1,5 @@
 import re
-from typing import TYPE_CHECKING, Any, Callable, Dict, Iterable, List, Optional, Tuple
+from typing import TYPE_CHECKING, Any, Callable, Dict, Iterable, List, Optional, Set, Tuple
 
 from .core import BlockState
 from .util import striptags, escape
@@ -68,11 +68,11 @@ def toc_hook(md: "Markdown", state: "BlockState") -> None:
     md.before_render_hooks.append(toc_hook)
 
 
-def _find_html_ids(src: str) -> set:
+def _find_html_ids(src: str) -> Set[str]:
     return {m.group(1) or m.group(2) for m in _HTML_ID_RE.finditer(src)}
 
 
-def _unique_id(value: str, used_ids: set) -> str:
+def _unique_id(value: str, used_ids: Set[str]) -> str:
     if value not in used_ids:
         return value