fix accidentally deleted lines

MannLabs · May 27, 2024 · 396706e · 396706e
1 parent 2b1dcaf
commit 396706e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sparcscore/utils/vis.py b/src/sparcscore/utils/vis.py
@@ -54,7 +54,8 @@ def plot_segmentation_mask(
             segmentation = segmentation[mask_channel, :, :]
             image = channels[image_channel, :, :]
         else:
-            segmentation = segmentation[mask_channel, selehannels[image_channel, selection[0], selection[1]]
+            segmentation = segmentation[mask_channel, selection[0], selection[1]]
+            image = channels[image_channel, selection[0], selection[1]]
 
     #set background to np.nan so that its not visualized
     segmentation = np.where(segmentation == background, np.nan, segmentation)