ENH: Support passing zarr stores in directly

bnmajor · Dec 8, 2023 · 49c8b7c · 49c8b7c
1 parent a01054f
commit 49c8b7c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/itkwidgets/integrations/__init__.py b/itkwidgets/integrations/__init__.py
@@ -59,6 +59,9 @@ def _get_viewer_image(image, label=False):
         to_ngff_zarr(store, image, chunk_store=chunk_store)
         return store
 
+    if isinstance(image, zarr.storage.Store):
+        return image
+
     if HAVE_MULTISCALE_SPATIAL_IMAGE:
         from multiscale_spatial_image import MultiscaleSpatialImage
         if isinstance(image, MultiscaleSpatialImage):
@@ -185,6 +188,8 @@ def _detect_render_type(data, input_type) -> RenderType:
             return RenderType.POINT_SET
         else:
             return RenderType.IMAGE
+    elif isinstance(data, zarr.storage.Store):
+        return RenderType.IMAGE
     elif HAVE_ITK:
         import itk
         if isinstance(data, itk.Image):