Check if fid is in fiberNorms, not the array of values

Subaru-PFS · Jun 18, 2024 · aff6df1 · aff6df1
1 parent 7cc4efc
commit aff6df1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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):