Add forceUniquePairs config

lsst-ts · Apr 15, 2024 · 41fe47d · 41fe47d
1 parent 159ba1b
commit 41fe47d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/python/lsst/ts/wep/task/pairTask.py b/python/lsst/ts/wep/task/pairTask.py
@@ -74,6 +74,11 @@ class ExposurePairerConfig(pexConfig.Config):
         default=0.1,
     )
 
+    forceUniquePairs = pexConfig.Field[bool](
+        doc="If True, force each extra exposure to be paired with a unique intra exposure.",
+        default=True,
+    )
+
 
 class ExposurePairer(pipeBase.Task):
     ConfigClass = ExposurePairerConfig
@@ -168,6 +173,12 @@ def run(
                 out.append(
                     IntraExtraIdxPair(nearbyTable[nearest]["exposure"], row["exposure"])
                 )
+                if self.config.forceUniquePairs:
+                    idx = np.where(
+                        intraTable["exposure"] == nearbyTable[nearest]["exposure"]
+                    )[0]
+                    intraTable.remove_rows(idx)
+
         return out