prevent plt.show() triggering when show_plot=False

cyschneck · Apr 12, 2024 · 7d108af · 7d108af
1 parent a736f2a
commit 7d108af
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/centerline_width/plotDiagrams.py b/centerline_width/plotDiagrams.py
@@ -244,6 +244,7 @@ def plotCenterline(river_object: centerline_width.riverCenterline = None,
 
     plt.legend(loc="upper right")
     if show_plot: plt.show()
+    if not show_plot: plt.close()
     if save_plot_name: fig.savefig(save_plot_name)
 
 
@@ -391,4 +392,5 @@ def plotCenterlineWidth(river_object: centerline_width.riverCenterline = None,
 
     plt.legend(loc="upper right")
     if show_plot: plt.show()
+    if not show_plot: plt.close()
     if save_plot_name: fig.savefig(save_plot_name)