Slight edit for scenario where non zero samples and 0 channels

funkelab · Oct 23, 2023 · 1f33fdb · 1f33fdb
1 parent a730c99
commit 1f33fdb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/napari_cellulus/dataset.py b/src/napari_cellulus/dataset.py
@@ -94,7 +94,6 @@ def __setup_pipeline(self):
                 NapariImageSource(
                     self.layer, self.raw, raw_spec, self.spatial_dims
                 )
-                + gp.Unsqueeze([self.raw], 1)
                 + gp.RandomLocation()
                 + gp.Unsqueeze([self.raw], 1)
             )