init ImageArray with kwargs

czbiohub-sf · Nov 25, 2024 · b7b8adc · b7b8adc
1 parent 71ec62c
commit b7b8adc
Showing 1 changed file with 18 additions and 14 deletions.
diff --git a/iohub/ngff/nodes.py b/iohub/ngff/nodes.py
@@ -310,20 +310,24 @@ def close(self):
 class ImageArray(zarr.Array):
     """Container object for image stored as a zarr array (up to 5D)"""
 
-    def __init__(self, zarray: zarr.Array):
-        super().__init__(
-            store=zarray._store,
-            path=zarray._path,
-            read_only=zarray._read_only,
-            chunk_store=zarray._chunk_store,
-            synchronizer=zarray._synchronizer,
-            cache_metadata=zarray._cache_metadata,
-            cache_attrs=zarray._attrs.cache,
-            partial_decompress=zarray._partial_decompress,
-            write_empty_chunks=zarray._write_empty_chunks,
-            zarr_version=zarray._version,
-            meta_array=zarray._meta_array,
-        )
+    def __init__(self, zarray: zarr.Array = None, **kwargs):
+        if zarray is not None:
+            kwargs.update(
+                {
+                    "store": zarray._store,
+                    "path": zarray._path,
+                    "read_only": zarray._read_only,
+                    "chunk_store": zarray._chunk_store,
+                    "synchronizer": zarray._synchronizer,
+                    "cache_metadata": zarray._cache_metadata,
+                    "cache_attrs": zarray._attrs.cache,
+                    "partial_decompress": zarray._partial_decompress,
+                    "write_empty_chunks": zarray._write_empty_chunks,
+                    "zarr_version": zarray._version,
+                    "meta_array": zarray._meta_array,
+                }
+            )
+        super().__init__(**kwargs)
         self._get_dims()
 
     def _get_dims(self):