fin

Author: Jammy2211

autoarray/mask/mask_1d.py

@@ -21,6 +21,10 @@
 logger = logging.getLogger(__name__)
 
 
+class Mask1DKeys(Enum):
+    PIXSCA = "PIXSCA"
+    ORIGIN = "ORIGIN"
+
 class Mask1D(Mask):
     def __init__(
         self,
@@ -173,11 +177,8 @@ def header_dict(self) -> Dict:
         -------
         A dictionary containing the pixel scale of the mask, which can be output to a .fits file.
         """
-        class GridKeys(Enum):
-            PIXSCA = "PIXSCA"
-            ORIGIN = "ORIGIN"
 
         return {
-            GridKeys.PIXSCA: self.pixel_scales[0],
-            GridKeys.ORIGIN: self.origin[0],
+            Mask1DKeys.PIXSCA: self.pixel_scales[0],
+            Mask1DKeys.ORIGIN: self.origin[0],
         }

autoarray/mask/mask_2d.py

@@ -31,6 +31,13 @@
 logger = logging.getLogger(__name__)
 
 
+class Mask2DKeys(Enum):
+    PIXSCAY = "PIXSCAY"
+    PIXSCAX = "PIXSCAX"
+    ORIGINY = "ORIGINY"
+    ORIGINX = "ORIGINX"
+
+
 class Mask2D(Mask):
     # noinspection PyUnusedLocal
     def __init__(
@@ -718,17 +725,11 @@ def header_dict(self) -> Dict:
         A dictionary containing the pixel scale of the mask, which can be output to a .fits file.
         """
 
-        class GridKeys(Enum):
-            PIXSCAY = "PIXSCAY"
-            PIXSCAX = "PIXSCAX"
-            ORIGINY = "ORIGINY"
-            ORIGINX = "ORIGINX"
-
         return {
-            GridKeys.PIXSCAY: self.pixel_scales[0],
-            GridKeys.PIXSCAX: self.pixel_scales[1],
-            GridKeys.ORIGINY: self.origin[0],
-            GridKeys.ORIGINX: self.origin[1],
+            Mask2DKeys.PIXSCAY: self.pixel_scales[0],
+            Mask2DKeys.PIXSCAX: self.pixel_scales[1],
+            Mask2DKeys.ORIGINY: self.origin[0],
+            Mask2DKeys.ORIGINX: self.origin[1],
         }
 
     @property