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)
