Only return pair for existing visitInfos

lsst-ts · Apr 4, 2024 · 86be05f · 86be05f
1 parent b80d372
commit 86be05f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/python/lsst/ts/wep/task/cutOutDonutsScienceSensorTask.py b/python/lsst/ts/wep/task/cutOutDonutsScienceSensorTask.py
@@ -48,11 +48,6 @@
 class CutOutDonutsScienceSensorTaskConnections(
     CutOutDonutsBaseTaskConnections, dimensions=("detector", "instrument")
 ):
-    def __init__(self, *, config=None):
-        super().__init__(config=config)
-        if not config.pairer.target._needsPairTable:
-            del self.donut_visit_pair_table
-
     visitInfos = ct.Input(
         doc="Input exposure to make measurements on",
         dimensions=("exposure", "detector", "instrument"),
@@ -67,6 +62,11 @@ def __init__(self, *, config=None):
         name="donut_visit_pair_table",
     )
 
+    def __init__(self, *, config=None):
+        super().__init__(config=config)
+        if not config.pairer.target._needsPairTable:
+            del self.donut_visit_pair_table
+
 
 class CutOutDonutsScienceSensorTaskConfig(
     CutOutDonutsBaseTaskConfig,
@@ -122,7 +122,9 @@ def runQuantum(
         }
 
         if hasattr(inputRefs, "donut_visit_pair_table"):
-            pairs = self.pairer.run(butlerQC.get(inputRefs.donut_visit_pair_table))
+            pairs = self.pairer.run(
+                visitInfoDict, butlerQC.get(inputRefs.donut_visit_pair_table)
+            )
         else:
             pairs = self.pairer.run(visitInfoDict)
         for pair in pairs:

diff --git a/python/lsst/ts/wep/task/pairTask.py b/python/lsst/ts/wep/task/pairTask.py
@@ -180,12 +180,14 @@ class TablePairer(pipeBase.Task):
 
     def run(
         self,
+        visitInfos: typing.Dict[int, afwImage.VisitInfo],
         pairTable: Table,
     ) -> typing.List[IntraExtraIdxPair]:
         """
         Pair up the intra- and extra-focal exposures.
         """
         out = []
         for row in pairTable:
-            out.append(IntraExtraIdxPair(row["intra"], row["extra"]))
+            if row["intra"] in visitInfos and row["extra"] in visitInfos:
+                out.append(IntraExtraIdxPair(row["intra"], row["extra"]))
         return out