Merge pull request #33 from youandvern/update-figure-temp-file-memory

Fix potential memory bloat with unclosed temp files
youandvern · Dec 6, 2024 · 545798a · 545798a
2 parents 0b6a927 + bb8f3eb
commit 545798a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/efficalc/base_definitions/figure.py b/efficalc/base_definitions/figure.py
@@ -64,9 +64,9 @@ def __init__(self, figure, caption: str = None, full_width: bool = False):
 
     def load_image_data(self) -> bytes:
         """Loads the image data from the Matplotlib figure object as bytes."""
-        tmp_file = BytesIO()
-        self.figure.savefig(tmp_file, format="png")
-        return tmp_file.getvalue()
+        with BytesIO() as tmp_file:
+            self.figure.savefig(tmp_file, format="png")
+            return tmp_file.getvalue()
 
 
 class FigureFromFile(FigureBase):