Fix crashes when FLUX keyword is not set in header.

lsst · Dec 7, 2023 · 4345504 · 4345504
1 parent 2f39c65
commit 4345504
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/python/lsst/cp/pipe/ptc/cpExtractPtcTask.py b/python/lsst/cp/pipe/ptc/cpExtractPtcTask.py
@@ -469,7 +469,11 @@ def run(self, inputExp, inputDims, taskMetadata, inputPhotodiodeData=None):
         # 'expTime' can stand for exposure time, flux, or ID.
         for expTime in inputExp:
             exposures = inputExp[expTime]
-            if len(exposures) == 1:
+            if not np.isfinite(expTime):
+                self.log.warning("Illegal/missing %s found (%f). Dropping exposure %d",
+                                 self.config.matchExposuresType, expTime, exposures[0][1])
+                continue
+            elif len(exposures) == 1:
                 self.log.warning("Only one exposure found at %s %f. Dropping exposure %d.",
                                  self.config.matchExposuresType, expTime, exposures[0][1])
                 continue

diff --git a/python/lsst/cp/pipe/utils.py b/python/lsst/cp/pipe/utils.py
@@ -500,7 +500,14 @@ def arrangeFlatsByExpFlux(exposureList, exposureIdList, fluxKeyword):
     assert len(exposureList) == len(exposureIdList), "Different lengths for exp. list and exp. ID lists"
     for expRef, expId in zip(exposureList, exposureIdList):
         # Get flux from header, assuming it is in the metadata.
-        expFlux = expRef.get().getMetadata()[fluxKeyword]
+        try:
+            expFlux = expRef.get().getMetadata()[fluxKeyword]
+        except KeyError:
+            # If it's missing from the header, continue; it will
+            # be caught and rejected when pairing exposures.
+            expFlux = None
+        if expFlux is None:
+            expFlux = np.nan
         listAtExpFlux = flatsAtExpFlux.setdefault(expFlux, [])
         listAtExpFlux.append((expRef, expId))