diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py index a3d8e41..f3971c7 100644 --- a/src/wireviz/Harness.py +++ b/src/wireviz/Harness.py @@ -208,14 +208,14 @@ class Harness: if not (r1, r2) == (None, None): dot.edge(r1, r2) - apply_dot_tweaks(dot, self.tweak) - for mate in self.mates: color, dir, code_from, code_to = gv_edge_mate(mate) dot.attr("edge", color=color, style="dashed", dir=dir) dot.edge(code_from, code_to) + apply_dot_tweaks(dot, self.tweak) + return dot # cache for the GraphViz Graph object