From 0c8021822479438ed855ebe3ebf769a7cb2f5e22 Mon Sep 17 00:00:00 2001
From: Marco Varrone <marco.varrone@unil.ch>
Date: Tue, 28 Oct 2025 12:16:07 +0100
Subject: [PATCH] Load Xenium mask labels using Dask

---
 src/spatialdata_io/readers/xenium.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/spatialdata_io/readers/xenium.py b/src/spatialdata_io/readers/xenium.py
index 87100c08..1244ecd0 100644
--- 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
             )