diff --git a/brainglobe_heatmap/heatmaps.py b/brainglobe_heatmap/heatmaps.py index 05aef79..56be41c 100644 --- a/brainglobe_heatmap/heatmaps.py +++ b/brainglobe_heatmap/heatmaps.py @@ -259,7 +259,7 @@ def plot( if show_legend: ax.legend() - + if ax is None: plt.show()