Update spectra.py

JarronL · Sep 5, 2024 · 42e5ab9 · 42e5ab9
1 parent dd7dc9b
commit 42e5ab9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/webbpsf_ext/spectra.py b/webbpsf_ext/spectra.py
@@ -1178,7 +1178,7 @@ def model_IRexcess(self, x, sp=None):
 
         bb_flux = x[0] * self.bb_jy(wave/1e4, x[1]) * (wave/1e4)**x[2] / 1e17
         sp_bb = s_ext.ArraySpectrum(wave, bb_flux, fluxunits='Jy')
-        sp_bb.convert('photlam')
+        sp_bb.convert(sp.fluxunits)
 
         return sp + sp_bb
 
@@ -1222,7 +1222,7 @@ def func_resid(self, x, IR_excess=False, wlim=[0.1, 30], use_err=True):
         mod_interp = np.interp(wvals, sp_star.wave, sp_model.flux)
 
         # Normalize values so the residuals aren't super small/large
-        norm = np.mean(yvals)
+        norm = np.nanmedian(yvals)
 
         resid = (mod_interp - yvals)
         if use_err: resid /= evals