From 275a1903f891e62b3404c5afbfe71baf342b2737 Mon Sep 17 00:00:00 2001
From: rhoadesScholar <rhoadesj@hhmi.org>
Date: Wed, 11 Sep 2024 14:57:13 -0400
Subject: [PATCH] Add get_image_grid_from_dict function to utils/__init__.py

---
 src/cellmap_data/utils/__init__.py | 2 +-
 src/cellmap_data/utils/figs.py     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/cellmap_data/utils/__init__.py b/src/cellmap_data/utils/__init__.py
index ffbb676..4c03fbc 100644
--- a/src/cellmap_data/utils/__init__.py
+++ b/src/cellmap_data/utils/__init__.py
@@ -1,2 +1,2 @@
-from .figs import get_image_grid, get_image_dict
+from .figs import get_image_grid, get_image_dict, get_image_grid_from_dict
 from .dtype import torch_max_value
diff --git a/src/cellmap_data/utils/figs.py b/src/cellmap_data/utils/figs.py
index defa182..6248be2 100644
--- a/src/cellmap_data/utils/figs.py
+++ b/src/cellmap_data/utils/figs.py
@@ -128,6 +128,7 @@ def get_image_grid_numpy(
         data = np.frombuffer(buff.getvalue(), dtype=np.uint8)
     w, h = fig.canvas.get_width_height()
     im = data.reshape((int(h), int(w), -1))
+    plt.close("all")
     return im