diff --git a/mandala/viz.py b/mandala/viz.py index f78ea34..782caaf 100644 --- a/mandala/viz.py +++ b/mandala/viz.py @@ -8,8 +8,6 @@ import subprocess import webbrowser from typing import Literal -from graphviz import Source -from IPython import display if Config.has_pil: from PIL import Image @@ -424,6 +422,10 @@ def write_output( webbrowser.open(str(output_path)) return elif show_how == "inline": + from graphviz import Source + from IPython import display + + src = Source(dot_string) display.display(src) elif show_how == "none":