fix

Jammy2211 · Dec 16, 2024 · 6f01e62 · 6f01e62
1 parent 9aff29b
commit 6f01e62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autoarray/dataset/imaging/dataset.py b/autoarray/dataset/imaging/dataset.py
@@ -400,7 +400,7 @@ def apply_noise_scaling(
             data = np.where(np.invert(mask), 0.0, self.data.native)
         elif data_noise_sigma is not None:
             random_noise = np.random.normal(
-                loc=data_noise_mean, scale=data_noise_sigma, size=data.shape_native
+                loc=data_noise_mean, scale=data_noise_sigma, size=self.data.shape_native
             )
             data = np.where(mask_2d, random_noise, data.native)