Load Xenium mask labels using Dask

marcovarrone · marcovarrone · commit 0c8021822479 · 2025-10-28T12:16:07.000+01:00
diff --git a/src/spatialdata_io/readers/xenium.py b/src/spatialdata_io/readers/xenium.py
@@ -458,7 +458,7 @@ def _get_labels_and_indices_mapping(
 
         with zarr_open(str(tmpdir), mode="r") as z:
             # get the labels
-            masks = z["masks"][f"{mask_index}"][...]
+            masks = da.from_array(z["masks"][f"{mask_index}"])
             labels = Labels2DModel.parse(
                 masks, dims=("y", "x"), transformations={"global": Identity()}, **labels_models_kwargs
             )

Original file line number	Diff line number	Diff line change
`@@ -458,7 +458,7 @@ def _get_labels_and_indices_mapping(`
`458`	`458`
`459`	`459`	`with zarr_open(str(tmpdir), mode="r") as z:`
`460`	`460`	`# get the labels`
`461`		`- masks = z["masks"][f"{mask_index}"][...]`
	`461`	`+ masks = da.from_array(z["masks"][f"{mask_index}"])`
`462`	`462`	`labels = Labels2DModel.parse(`
`463`	`463`	`masks, dims=("y", "x"), transformations={"global": Identity()}, **labels_models_kwargs`
`464`	`464`	`)`