Add get_image_grid_from_dict function to utils/__init__.py

janelia-cellmap · Sep 11, 2024 · 275a190 · 275a190
1 parent eb4f76b
commit 275a190
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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