From 0cb2de5d0634c313ca2a98dd9114ac3a10a1bab4 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 15:19:31 +0100 Subject: [PATCH] train-preedict: close matplotlib after we've finished they act like file handles --- aimodel/src/lib/vis/embeddings.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/vis/embeddings.py b/aimodel/src/lib/vis/embeddings.py index a0698b7..0df6aa2 100644 --- a/aimodel/src/lib/vis/embeddings.py +++ b/aimodel/src/lib/vis/embeddings.py @@ -42,4 +42,5 @@ def vis_embeddings(filepath_output, features): axes["B"].set_title(f"Parallel coordinates plot", fontsize=20) plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", 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