Fix labels boundaries (#281)

melonora · web-flow · commit d2618e70ab67 · 2024-06-23T13:07:10.000+02:00
* implement label fix

* fix label boundaries
diff --git a/src/spatialdata_plot/pl/utils.py b/src/spatialdata_plot/pl/utils.py
@@ -775,6 +775,8 @@ def _map_color_seg(
     )
 
     if seg_boundaries:
+        if seg.shape[0] == 1:
+            seg = np.squeeze(seg, axis=0)
         seg_bound: ArrayLike = np.clip(seg_im - find_boundaries(seg)[:, :, None], 0, 1)
         return np.dstack((seg_bound, np.where(val_im > 0, 1, 0)))  # add transparency here
 

Original file line number	Diff line number	Diff line change
`@@ -775,6 +775,8 @@ def _map_color_seg(`
`775`	`775`	`)`
`776`	`776`
`777`	`777`	`if seg_boundaries:`
	`778`	`+ if seg.shape[0] == 1:`
	`779`	`+ seg = np.squeeze(seg, axis=0)`
`778`	`780`	`seg_bound: ArrayLike = np.clip(seg_im - find_boundaries(seg)[:, :, None], 0, 1)`
`779`	`781`	`return np.dstack((seg_bound, np.where(val_im > 0, 1, 0))) # add transparency here`
`780`	`782`