Skip to content

Commit 04bdc71

Browse files
committed
sort the nodes, edges, and plates
1 parent 174e72f commit 04bdc71

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pymc/model_graph.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -906,9 +906,9 @@ def model_to_mermaid(model=None, *, var_names=None, include_dim_lengths: bool =
906906
"""
907907
model = pm.modelcontext(model)
908908
graph = ModelGraph(model)
909-
plates = graph.get_plates(var_names=var_names)
910-
edges = graph.edges(var_names=var_names)
911-
nodes = graph.nodes(plates=plates)
909+
plates = sorted(graph.get_plates(var_names=var_names), key=lambda plate: hash(plate.dim_info))
910+
edges = sorted(graph.edges(var_names=var_names))
911+
nodes = sorted(graph.nodes(plates=plates), key=lambda node: node.var.name)
912912

913913
return "\n".join(
914914
[

0 commit comments

Comments
 (0)