Use the correct image for alternate spatial bases

obriencshl · web-flow · commit 531bd21f85df · 2025-09-08T16:19:40.000-04:00
Retrieve the image from adata.uns[spatial_key] rather than always adata.uns["spatial"].
diff --git a/src/squidpy/pl/_spatial_utils.py b/src/squidpy/pl/_spatial_utils.py
@@ -171,7 +171,7 @@ def _get_image(
             raise KeyError(
                 f"Image key: `{img_res_key}` does not exist. Available image keys: `{image_mapping.values()}`"
             )
-        img = [adata.uns[Key.uns.spatial][i][Key.uns.image_key][img_res_key] for i in library_id]
+        img = [adata.uns[spatial_key][i][Key.uns.image_key][img_res_key] for i in library_id]
 
     if img_channel is None:
         img = [im[..., :3] for im in img]

Original file line number	Diff line number	Diff line change
`@@ -171,7 +171,7 @@ def _get_image(`
`171`	`171`	`raise KeyError(`
`172`	`172`	f"Image key: `{img_res_key}` does not exist. Available image keys: `{image_mapping.values()}`"
`173`	`173`	`)`
`174`		`- img = [adata.uns[Key.uns.spatial][i][Key.uns.image_key][img_res_key] for i in library_id]`
	`174`	`+ img = [adata.uns[spatial_key][i][Key.uns.image_key][img_res_key] for i in library_id]`
`175`	`175`
`176`	`176`	`if img_channel is None:`
`177`	`177`	`img = [im[..., :3] for im in img]`