notebook rendering

transferwise · May 14, 2024 · 56b7492 · 56b7492
1 parent 0f4b7ed
commit 56b7492
Show file tree

Hide file tree

Showing 2 changed files with 124 additions and 127 deletions.
diff --git a/notebooks/Finding interesting segments.ipynb b/notebooks/Finding interesting segments.ipynb
diff --git a/wise_pizza/plotting.py b/wise_pizza/plotting.py
@@ -139,14 +139,21 @@ def plot_split_segments(
         image_bytes = to_image(fig, format="png", scale=2)
 
         if sf_size.relevant_cluster_names:
+            image_bytes2 = to_image(fig2, format="png", scale=2)
             display(
                 Image(
                     image_bytes,
                     height=height + len(size_data.index) * 30,
                     width=width + len(size_data.index) * 30,
                 )
             )
-            fig2.show()
+            display(
+                Image(
+                    image_bytes2,
+                    height=height + len(size_data.index) * 30,
+                    width=width + len(size_data.index) * 30,
+                )
+            )
 
         else:
             # Display the static image in the Jupyter notebook
@@ -410,8 +417,9 @@ def plot_waterfall(
         # Convert the figure to a static image
         image_bytes = to_image(fig, format="png", scale=2)
         if sf.relevant_cluster_names:
+            image_bytes2 = to_image(fig2, format="png", scale=2)
             display(Image(image_bytes, height=height, width=width))
-            fig2.show()
+            display(Image(image_bytes2, height=height, width=width))
         else:
             # Display the static image in the Jupyter notebook
             display(Image(image_bytes, width=width, height=height))