Fix bug where a new colorbar figure would appear on each valid scan

2026-05-01 00:09:12 -05:00
parent 2ed37f95a7
commit a74ca01287
1 changed files with 4 additions and 0 deletions
@@ -62,6 +62,10 @@ def render_heatmap(
    if fig is None or ax is None:
        fig, ax = plt.subplots(figsize=(10, 7))

+    # Remove any colorbars left over from previous renders.
+    # Colorbars attach to the figure, not the axes, so ax.clear() doesn't touch them.
+    for ax_obj in fig.axes[1:]:   # axes[0] is the main plot; everything else is a colorbar
+        ax_obj.remove()
    ax.clear()

    points, values = session.get_points_and_values_multi(bssids)