Implement new connections hook for refcat lookups.

This is part of a new, slightly-better fix for DM-40243.
lsst · Aug 6, 2023 · 402a410 · 402a410
1 parent cee6e8c
commit 402a410
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 0 deletions.
diff --git a/python/lsst/fgcmcal/fgcmBuildFromIsolatedStars.py b/python/lsst/fgcmcal/fgcmBuildFromIsolatedStars.py
@@ -138,6 +138,9 @@ def __init__(self, *, config=None):
             self.prerequisiteInputs.remove("fgcm_lookup_table")
             self.outputs.remove("fgcm_reference_stars")
 
+    def getSpatialBoundsConnections(self):
+        return ("isolated_star_cats", "visit_summaries")
+
 
 class FgcmBuildFromIsolatedStarsConfig(FgcmBuildStarsConfigBase, pipeBase.PipelineTaskConfig,
                                        pipelineConnections=FgcmBuildFromIsolatedStarsConnections):

diff --git a/python/lsst/fgcmcal/fgcmBuildStarsTable.py b/python/lsst/fgcmcal/fgcmBuildStarsTable.py
@@ -149,6 +149,9 @@ def __init__(self, *, config=None):
         if not config.doReferenceMatches:
             self.outputs.remove("fgcmReferenceStars")
 
+    def getSpatialBoundsConnections(self):
+        return ("visit_summaries",)
+
 
 class FgcmBuildStarsTableConfig(FgcmBuildStarsConfigBase, pipeBase.PipelineTaskConfig,
                                 pipelineConnections=FgcmBuildStarsTableConnections):

diff --git a/python/lsst/fgcmcal/fgcmCalibrateTractTable.py b/python/lsst/fgcmcal/fgcmCalibrateTractTable.py
@@ -135,6 +135,9 @@ def __init__(self, *, config=None):
         if not config.fgcmOutputProducts.doZeropointOutput:
             self.prerequisiteInputs.remove("fgcmZeropoints")
 
+    def getSpatialBoundsConnections(self):
+        return ("isolated_star_cats", "visit_summaries")
+
 
 class FgcmCalibrateTractTableConfig(FgcmCalibrateTractConfigBase, pipeBase.PipelineTaskConfig,
                                     pipelineConnections=FgcmCalibrateTractTableConnections):

diff --git a/python/lsst/fgcmcal/fgcmOutputProducts.py b/python/lsst/fgcmcal/fgcmOutputProducts.py
@@ -163,6 +163,9 @@ def __init__(self, *, config=None):
         if not config.doReferenceCalibration:
             self.outputs.remove("fgcmOffsets")
 
+    def getSpatialBoundsConnections(self):
+        return ("fgcmPhotoCalibCatalog",)
+
 
 class FgcmOutputProductsConfig(pipeBase.PipelineTaskConfig,
                                pipelineConnections=FgcmOutputProductsConnections):