move should zero data

Author: Jammy2211

autoarray/dataset/imaging/dataset.py

@@ -383,6 +383,16 @@ def apply_noise_scaling(
             If True, the data values in the masked region are set to zero.
         """
 
+        if signal_to_noise_value is None:
+            noise_map = self.noise_map.native
+            noise_map[mask == False] = noise_value
+        else:
+            noise_map = np.where(
+                mask == False,
+                np.abs(self.data.native) / signal_to_noise_value,
+                self.noise_map.native,
+            )
+
         if should_zero_data:
             data = np.where(np.invert(mask), 0.0, self.data.native)
         else:
@@ -394,16 +404,6 @@ def apply_noise_scaling(
             pixel_scales=self.data.pixel_scales,
         )
 
-        if signal_to_noise_value is None:
-            noise_map = self.noise_map.native
-            noise_map[mask == False] = noise_value
-        else:
-            noise_map = np.where(
-                mask == False,
-                np.abs(data.native) / signal_to_noise_value,
-                self.noise_map.native,
-            )
-
         noise_map = Array2D.no_mask(
             values=noise_map,
             shape_native=self.data.shape_native,