disallow pixels with zero flux error

sdss · Sep 30, 2022 · bb864f2 · bb864f2
1 parent a2d487c
commit bb864f2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/python/astra/sdss/datamodels/apogee.py b/python/astra/sdss/datamodels/apogee.py
@@ -124,6 +124,13 @@ def create_apogee_hdus(
 
     wavelength = util.log_lambda_dispersion(crval, cdelt, num_pixels)
 
+    # Disallow zero fluxes.
+    bad_flux, bad_combined_flux = ((flux_error == 0), (combined_flux_error == 0))
+    flux[bad_flux] = np.nan
+    flux_error[bad_flux] = np.inf
+    combined_flux[bad_combined_flux] = np.nan
+    combined_flux_error[bad_combined_flux] = np.inf
+
     DATA_HEADER_CARD = ("SPECTRAL DATA", None)
 
     visit_mappings = [

diff --git a/python/astra/sdss/datamodels/boss.py b/python/astra/sdss/datamodels/boss.py
@@ -154,6 +154,13 @@ def create_boss_hdus(
     meta.update(meta_combine)
     wavelength = util.log_lambda_dispersion(crval, cdelt, num_pixels)
 
+    # Disallow zero fluxes.
+    bad_flux, bad_combined_flux = ((flux_error == 0), (combined_flux_error == 0))
+    flux[bad_flux] = np.nan
+    flux_error[bad_flux] = np.inf
+    combined_flux[bad_combined_flux] = np.nan
+    combined_flux_error[bad_combined_flux] = np.inf
+
     DATA_HEADER_CARD = ("SPECTRAL DATA", None)
 
     nanify = lambda x: np.nan if x == "NaN" else x

diff --git a/python/astra/sdss/datamodels/combine.py b/python/astra/sdss/datamodels/combine.py
@@ -276,8 +276,6 @@ def resample_visit_spectra(
                 np.uint64
             )
 
-
-
     return (
         resampled_flux,
         resampled_flux_error,