Replace < and >

sdv-dev · Nov 7, 2023 · 3f5c327 · 3f5c327
1 parent 3e0de8e
commit 3f5c327
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sdv/metadata/visualization.py b/sdv/metadata/visualization.py
@@ -105,6 +105,7 @@ def visualize_graph(nodes, edges, filepath=None):
     )
 
     for name, label in nodes.items():
+        label = label.replace('<', 'lt').replace('>', 'gt')
         digraph.node(name, label=label)
 
     for parent, child, label in edges: