From a3c145492a25067c74e07ee013acc5b542b06533 Mon Sep 17 00:00:00 2001
From: James Nightingale <james.w.nightingale@durham.ac.uk>
Date: Wed, 15 Jan 2025 21:12:56 +0000
Subject: [PATCH] final clean up

---
 autoarray/dataset/imaging/dataset.py | 14 ++++----------
 1 file changed, 4 insertions(+), 10 deletions(-)

diff --git a/autoarray/dataset/imaging/dataset.py b/autoarray/dataset/imaging/dataset.py
index 7c7b01a9..78449ee4 100644
--- a/autoarray/dataset/imaging/dataset.py
+++ b/autoarray/dataset/imaging/dataset.py
@@ -443,24 +443,18 @@ def apply_noise_scaling(
         data_unmasked = Array2D.no_mask(
             values=data,
             shape_native=self.data.shape_native,
-            pixel_scales=self.data.pixel_scales
+            pixel_scales=self.data.pixel_scales,
         )
 
         noise_map_unmasked = Array2D.no_mask(
             values=noise_map,
             shape_native=self.noise_map.shape_native,
-            pixel_scales=self.noise_map.pixel_scales
+            pixel_scales=self.noise_map.pixel_scales,
         )
 
-        data = Array2D(
-            values=data,
-            mask=self.data.mask
-        )
+        data = Array2D(values=data, mask=self.data.mask)
 
-        noise_map = Array2D(
-            values=noise_map,
-            mask=self.data.mask
-        )
+        noise_map = Array2D(values=noise_map, mask=self.data.mask)
 
         dataset = Imaging(
             data=data,