Skip to content

Commit cfee369

Browse files
committed
move should zero data
1 parent 824bf07 commit cfee369

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

autoarray/dataset/imaging/dataset.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -383,6 +383,16 @@ def apply_noise_scaling(
383383
If True, the data values in the masked region are set to zero.
384384
"""
385385

386+
if signal_to_noise_value is None:
387+
noise_map = self.noise_map.native
388+
noise_map[mask == False] = noise_value
389+
else:
390+
noise_map = np.where(
391+
mask == False,
392+
np.abs(self.data.native) / signal_to_noise_value,
393+
self.noise_map.native,
394+
)
395+
386396
if should_zero_data:
387397
data = np.where(np.invert(mask), 0.0, self.data.native)
388398
else:
@@ -394,16 +404,6 @@ def apply_noise_scaling(
394404
pixel_scales=self.data.pixel_scales,
395405
)
396406

397-
if signal_to_noise_value is None:
398-
noise_map = self.noise_map.native
399-
noise_map[mask == False] = noise_value
400-
else:
401-
noise_map = np.where(
402-
mask == False,
403-
np.abs(data.native) / signal_to_noise_value,
404-
self.noise_map.native,
405-
)
406-
407407
noise_map = Array2D.no_mask(
408408
values=noise_map,
409409
shape_native=self.data.shape_native,

0 commit comments

Comments
 (0)