phowell
/
pocketflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mermaid version issue. use simple label instead

This commit is contained in:
zachary62 2025-01-07 02:44:48 +00:00
parent eacf6a3d70
commit beafd5bc8b
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
docs/viz.md
View File

@ -58,10 +58,7 @@ def build_mermaid(start):
else: else:
# Handle simple Nodes # Handle simple Nodes
curr_id = get_id(node) curr_id = get_id(node)
if isinstance(node, BatchNode): label = f'{curr_id}["{type(node).__name__}"]'
label = f'{curr_id}@{{shape: procs, label: "{type(node).__name__}"}}'
else:
label = f'{curr_id}@{{label: "{type(node).__name__}"}}'
if parent_id: if parent_id:
lines.append(f" {label}") lines.append(f" {label}")
@ -116,16 +113,16 @@ The above code produces a Mermaid diagram (e.g., use the [Mermaid Live Editor](h
```mermaid ```mermaid
graph LR graph LR
N1@{shape: procs, label: "DataPrepBatchNode"} N1["DataPrepBatchNode"]
N2@{label: "ValidateDataNode"} N2["ValidateDataNode"]
N1 --> N2 N1 --> N2
N2 --> N3 N2 --> N3
subgraph sub_flow_N4[ModelFlow] subgraph sub_flow_N4[ModelFlow]
N3@{label: "FeatureExtractionNode"} N3["FeatureExtractionNode"]
N5@{label: "TrainModelNode"} N5["TrainModelNode"]
N3 --> N5 N3 --> N5
N6@{label: "EvaluateModelNode"} N6["EvaluateModelNode"]
N5 --> N6 N5 --> N6
end end
``` ```