Merge pull request #568 from opencybersecurityalliance/k2-apply-explain

minor jupyter display fix
opencybersecurityalliance · Jul 28, 2024 · eebe248 · eebe248
2 parents f43e927 + acdeac9
commit eebe248
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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'<img src="data:image/png;base64,{img}">'
-            yield imgx
+            img_base64 = base64.b64encode(fig_buffer.getvalue()).decode("utf-8")
+            img_tag = f'<img src="data:image/png;base64,{img_base64}">'
+            yield img_tag
 
             if isinstance(graphlet.action, NativeQuery):
                 native_query = graphlet.action