fix bug in putting noise maps in 4d

samuelstjean · Mar 9, 2021 · a2b9c52 · a2b9c52
1 parent b427775
commit a2b9c52
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/get_distribution b/scripts/get_distribution
@@ -152,8 +152,8 @@ def main():
         else:
             overlap = 'non overlapping windows'
 
-        if noise_maps.ndim == 3:
-            noise_maps = noise_maps[..., None]
+        if data.ndim == 3:
+            data = data[..., None]
 
         logger.info('Estimation will be done over noise maps with a window of size {} and {}'.format(size, overlap))
         sigma, N, mask = estimate_from_nmaps(data, size=size, return_mask=True, method=method, full=full, ncores=ncores, use_rejection=False, verbose=args.verbose)