yobx.translate.mermaid_emitter#

class yobx.translate.mermaid_emitter.MermaidEmitter(edge_labels: Dict[str, str] | None = None)[source]#

Emitter that converts an ONNX graph into a Mermaid flowchart TD string.

Intended to be used with Translator:

from yobx.translate.mermaid_helper import MermaidEmitter
from yobx.translate.translator import Translator

emitter = MermaidEmitter()
tr = Translator(model, emitter=emitter)
print(tr.export(as_str=True))

Node colour legend:

  • green (classDef input) – graph inputs

  • yellow (classDef init) – initializers / constant weights

  • light-grey (classDef op) – ONNX operators

  • light-blue (classDef output) – graph outputs

join(rows: List[str], single_line: bool = False) → str[source]#

Join Mermaid lines into a single string.