diff --git a/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py b/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py index 321cf9b8..2c7ba38f 100644 --- a/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py +++ b/packages/kestrel_jupyter/src/kestrel_jupyter_kernel/display.py @@ -54,9 +54,9 @@ def to_html_blocks(d: Display) -> Iterable[str]: ) fig_buffer = BytesIO() plt.savefig(fig_buffer, format="png") - img = data_uri = base64.b64encode(fig_buffer.getvalue()).decode("utf-8") - imgx = f'' - yield imgx + img_base64 = base64.b64encode(fig_buffer.getvalue()).decode("utf-8") + img_tag = f'' + yield img_tag if isinstance(graphlet.action, NativeQuery): native_query = graphlet.action