Exposure time can also be exposure map as a regular pixealated grid

aymgal · Nov 13, 2023 · b85dd24 · b85dd24
1 parent 9b5bd63
commit b85dd24
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/coolest/template/classes/observation.py b/coolest/template/classes/observation.py
@@ -1,8 +1,6 @@
 __author__ = 'aymgal'
 
-from astropy.io import fits
-import numpy as np
-from typing import Tuple
+from typing import Union
 
 from coolest.template.classes.grid import PixelatedRegularGrid
 from coolest.template.classes.noise import Noise
@@ -31,7 +29,7 @@ class Observation(APIBaseObject):
 
     def __init__(self, 
                  pixels: PixelatedRegularGrid = None,  # can be None for mock generation
-                 exposure_time: float = None,
+                 exposure_time: Union[float, PixelatedRegularGrid] = None,
                  noise: Noise = None,
                  mag_zero_point: float = None,
                  mag_sky_brightness: float = None,