Fixed node highlighting for single highlight argument

faustedition · Apr 17, 2019 · 9c1583b · 9c1583b
1 parent bd7e741
commit 9c1583b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/macrogen/visualize.py b/src/macrogen/visualize.py
@@ -119,7 +119,7 @@ def write_dot(graph: nx.MultiDiGraph, target: Union[PathLike, str] = 'base_graph
     if highlight is not None:
         if not isinstance(highlight, Sequence):
             highlight = [highlight]
-        elif 'highlight' in style['node']:
+        if 'highlight' in style['node']:
             for highlight_node in highlight:
                 try:
                     vis.nodes[highlight_node].update(style['node']['highlight'])