Simplify code

Author: zaxtax

pymc/model_graph.py

@@ -430,9 +430,6 @@ def edges(
         ]
 
 
-GraphAttrMapping = dict[Any, Any]
-
-
 def make_graph(
     name: str,
     plates: list[Plate],
@@ -442,7 +439,7 @@ def make_graph(
     figsize=None,
     dpi=300,
     node_formatters: NodeTypeFormatterMapping | None = None,
-    graph_attrs: GraphAttrMapping | None = None,
+    graph_attr: dict[str, Any] | None = None,
     create_plate_label: PlateLabelFunc = create_plate_label_with_dim_length,
 ):
     """Make graphviz Digraph of PyMC model.
@@ -463,7 +460,7 @@ def make_graph(
     node_formatters = node_formatters or {}
     node_formatters = update_node_formatters(node_formatters)
 
-    graph = graphviz.Digraph(name, graph_attr=graph_attrs)
+    graph = graphviz.Digraph(name, graph_attr=graph_attr)
     for plate in plates:
         if plate.dim_info:
             # must be preceded by 'cluster' to get a box around it
@@ -680,7 +677,7 @@ def model_to_graphviz(
     figsize: tuple[int, int] | None = None,
     dpi: int = 300,
     node_formatters: NodeTypeFormatterMapping | None = None,
-    graph_attrs: GraphAttrMapping | None = None,
+    graph_attr: dict[str, Any] | None = None,
     include_dim_lengths: bool = True,
 ):
     """Produce a graphviz Digraph from a PyMC model.
@@ -709,7 +706,7 @@ def model_to_graphviz(
         the size of the saved figure.
     dpi : int, optional
         Dots per inch. It only affects the resolution of the saved figure. The default is 300.
-    graph_attrs : dict, optional
+    graph_attr : dict, optional
         A dictionary of top-level layout attributes for graphviz
         Check out graphviz documentation for more information on available attributes
         https://graphviz.org/doc/info/attrs.html
@@ -782,7 +779,7 @@ def model_to_graphviz(
         save=save,
         figsize=figsize,
         dpi=dpi,
-        graph_attrs=graph_attrs,
+        graph_attr=graph_attr,
         node_formatters=node_formatters,
         create_plate_label=create_plate_label_with_dim_length
         if include_dim_lengths