From 42aea7a0cc34b5e9b960e37994a33cb9885f4472 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 15:23:54 +0100 Subject: [PATCH] plt.close() fixup --- aimodel/src/lib/vis/segmentation_plot.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/vis/segmentation_plot.py b/aimodel/src/lib/vis/segmentation_plot.py index a018f03..aca3165 100644 --- a/aimodel/src/lib/vis/segmentation_plot.py +++ b/aimodel/src/lib/vis/segmentation_plot.py @@ -25,4 +25,5 @@ def segmentation_plot(water_actual, water_predict, model_code, filepath_output): plt.suptitle(f"Rainfall → Water depth prediction | {model_code}", fontsize=28, weight="bold") - plt.savefig(filepath_output) \ No newline at end of file + plt.savefig(filepath_output) + plt.close() \ No newline at end of file