def generate_image(graph: pydot.Dot, red_nodes: List[str]) -> str: for node in red_nodes: node = graph.get_node(node)[0] node.set_color("#d35d6e") _, png = mkstemp() with open(png, "wb") as f: f.write(graph.create_png()) print(f"Wrote intermediate step to {png}") return png
def plot_dot(pdot: pydot.Dot) -> None: """ Displays a DOT object in the notebook """ plt = Image(pdot.create_png()) display(plt)