Lint fix

Author: zaxtax

pymc/model_graph.py

@@ -430,6 +430,8 @@ def edges(
         ]
 
 
+GraphAttrMapping = dict[Any, Any]
+
 def make_graph(
     name: str,
     plates: list[Plate],
@@ -439,7 +441,7 @@ def make_graph(
     figsize=None,
     dpi=300,
     node_formatters: NodeTypeFormatterMapping | None = None,
-    graph_attrs : dict[str, Any] | None,
+    graph_attrs : GraphAttrMapping | None = None,
     create_plate_label: PlateLabelFunc = create_plate_label_with_dim_length,
 ):
     """Make graphviz Digraph of PyMC model.
@@ -680,7 +682,7 @@ def model_to_graphviz(
     figsize: tuple[int, int] | None = None,
     dpi: int = 300,
     node_formatters: NodeTypeFormatterMapping | None = None,
-    graph_attrs: dict[str, Any] | None = None,
+    graph_attrs: GraphAttrMapping | None = None,
     include_dim_lengths: bool = True,
 ):
     """Produce a graphviz Digraph from a PyMC model.