diff --git a/python/pfs/drp/stella/extractSpectraTask.py b/python/pfs/drp/stella/extractSpectraTask.py index 0c86f7424..7a886d20e 100644 --- a/python/pfs/drp/stella/extractSpectraTask.py +++ b/python/pfs/drp/stella/extractSpectraTask.py @@ -160,7 +160,7 @@ def extractAllSpectra( if fiberNorms is not None: for spectrum in spectra: - if spectrum.fiberId in fiberNorms.coeff: + if spectrum.fiberId in fiberNorms: spectrum.norm *= fiberNorms.calculate(spectrum.fiberId) return spectra diff --git a/python/pfs/drp/stella/photometerLines.py b/python/pfs/drp/stella/photometerLines.py index 9c1bd2b05..f02f59a26 100644 --- a/python/pfs/drp/stella/photometerLines.py +++ b/python/pfs/drp/stella/photometerLines.py @@ -233,7 +233,7 @@ def addFluxNormalizations(self, lines, tracesOrProfiles, fiberNorms): ] if fiberNorms is not None: for i, (ff, yy) in enumerate(zip(lines.fiberId, lines.y)): - if ff in fiberNorms.coeff: + if ff in fiberNorms: lines.fluxNorm[i] = fiberNorms.calculate(ff, yy) def subtractLines(self, exposure, lines, apCorr, pfsConfig):