passing mypy locally

Author: chrishavlin

inheritance_explorer/inheritance_explorer.py

@@ -25,7 +25,7 @@ def __init__(
         color: Optional[str] = "#000000",
     ):
         self.child = child
-        self.child_name = child.__name__
+        self.child_name: str = str(child.__name__)
         self._child_id = child_id
         self.parent = parent
 
@@ -93,14 +93,14 @@ def __init__(
         self._tracking_function = self.funcname is not None
         self.max_recursion_level = max_recursion_level
         self._nodenum: int = 0
-        self._node_list = []  # a list of unique ChildNodes
-        self._node_map = {}  # map of global node index to node name
+        self._node_list: list[_ChildNode] = []  # a list of unique ChildNodes
+        self._node_map: dict[int, str] = {}  # map of global node index to node name
         self._override_src: OrderedDict[int, str] = collections.OrderedDict()
-        self._override_src_files = {}
+        self._override_src_files: dict[int, str] = {}
         self._current_node = 1  # the current global node, must start at 1
         self._default_color = default_color
         self._override_color = func_override_color
-        self._graphviz_args_kwargs = {}
+        self._graphviz_args_kwargs: dict[str, Any] = {}
         self.similarity_container: _similarity_container_types | None = None
         self.similarity_results: dict[str, npt.NDArray[Any]]
         self.similarity_cutoff = similarity_cutoff
@@ -314,7 +314,7 @@ def plot_similarity(
         ax: Optional[Axes] = None,
         colorbar: Optional[bool] = True,
         **kwargs,
-    ) -> tuple[dict, Axes]:
+    ) -> tuple[dict[int, str], Axes]:
         """
         add the similarity plot to a matplotlib axis (or create a new one)
 
@@ -371,7 +371,7 @@ def build_interactive_graph(
         node_style: dict[str, Any] | None = None,
         edge_style: dict[str, Any] | None = None,
         similarity_edge_style: dict[str, Any] | None = None,
-        override_node_color: str | tuple | None = None,
+        override_node_color: str | tuple[float, ...] | None = None,
         **kwargs,
     ) -> Network:
         """