handy dandy cast

williambdean · williambdean · commit a7dd63a28118 · 2025-06-19T16:30:51.000-04:00
diff --git a/pymc/model_graph.py b/pymc/model_graph.py
@@ -908,7 +908,7 @@ def model_to_mermaid(model=None, *, var_names=None, include_dim_lengths: bool =
     graph = ModelGraph(model)
     plates = sorted(graph.get_plates(var_names=var_names), key=lambda plate: hash(plate.dim_info))
     edges = sorted(graph.edges(var_names=var_names))
-    nodes = sorted(graph.nodes(plates=plates), key=lambda node: node.var.name)
+    nodes = sorted(graph.nodes(plates=plates), key=lambda node: cast(str, node.var.name))
 
     return "\n".join(
         [

Original file line number	Diff line number	Diff line change
`@@ -908,7 +908,7 @@ def model_to_mermaid(model=None, *, var_names=None, include_dim_lengths: bool =`
`908`	`908`	`graph = ModelGraph(model)`
`909`	`909`	`plates = sorted(graph.get_plates(var_names=var_names), key=lambda plate: hash(plate.dim_info))`
`910`	`910`	`edges = sorted(graph.edges(var_names=var_names))`
`911`		`- nodes = sorted(graph.nodes(plates=plates), key=lambda node: node.var.name)`
	`911`	`+ nodes = sorted(graph.nodes(plates=plates), key=lambda node: cast(str, node.var.name))`
`912`	`912`
`913`	`913`	`return "\n".join(`
`914`	`914`	`[`