Commit 158e8d2e authored by Daniel Brown's avatar Daniel Brown

Moving graphviz call into method so it only ever gets loaded when it's called

parent a03f4123
......@@ -5,8 +5,6 @@ Based on node_graph project by Sebastian Steinlechner:
import tempfile
from ...components import laser, space
from graphviz import Digraph
class NodeGraph(object):
"""Pykat node graph plotter."""
......@@ -27,6 +25,7 @@ class NodeGraph(object):
graph_font_size=8, component_color="LightSkyBlue", laser_color="Orange",
space_color="MediumSlateBlue", detector_color="YellowGreen"):
"""Create Graphviz node graph"""
from graphviz import Digraph
graph = Digraph(engine=engine)
graph.attr("node", style=node_style, fontname=node_font, fontsize=str(node_font_size))
graph.attr("edge", arrowhead=edge_arrowhead)
......@@ -7,3 +7,4 @@ pandas
click >= 7.0
\ No newline at end of file
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment