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)