diff --git a/src/extraction/core/extractor.py b/src/extraction/core/extractor.py index 4fc5e484831f75e9f7b9c15ce200d1ccfdcf62f6..3f7fdbef8c680e1902f28b84022eb6cf4f02175e 100644 --- a/src/extraction/core/extractor.py +++ b/src/extraction/core/extractor.py @@ -482,7 +482,7 @@ class Extractor(StepABC): # stored as an array arranged as (traps, channels, time points, X, Y, Z) tiles = self.get_tiles(tp, tile_shape=tile_size, channels=tree_chs) # generate boolean masks for background as a list with one mask per trap - bgs = [] + bgs = np.array([]) if self.params.sub_bg: # bgs = [ # ~np.sum(m, axis=0).astype(bool)