re-introduce uncertainty fallbacks for HorneExtract

spacetelescope · Dec 8, 2022 · c371981 · c371981
1 parent 9e2aedc
commit c371981
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/jdaviz/configs/specviz2d/plugins/spectral_extraction/spectral_extraction.py b/jdaviz/configs/specviz2d/plugins/spectral_extraction/spectral_extraction.py
@@ -14,7 +14,7 @@
 from jdaviz.core.marks import PluginLine
 
 from astropy.modeling import models
-from astropy.nddata import UnknownUncertainty
+from astropy.nddata import NDData, StdDevUncertainty, VarianceUncertainty, UnknownUncertainty
 from astropy import units
 from specreduce import tracing
 from specreduce import background
@@ -884,6 +884,10 @@ def export_extract(self, **kwargs):
         if self.ext_type_selected == 'Boxcar':
             ext = extract.BoxcarExtract(inp_sp2d, trace, width=self.ext_width)
         elif self.ext_type_selected == 'Horne':
+            if inp_sp2d.uncertainty is None:
+                inp_sp2d.uncertainty = VarianceUncertainty(np.ones_like(inp_sp2d.data))
+            if not hasattr(inp_sp2d.uncertainty, 'uncertainty_type'):
+                inp_sp2d.uncertainty = StdDevUncertainty(inp_sp2d.uncert)
             ext = extract.HorneExtract(inp_sp2d, trace)
         else:
             raise NotImplementedError(f"extraction type '{self.ext_type_selected}' not supported")  # noqa