Merge branch 'tickets/DM-42157'

Author: taranu

python/lsst/pipe/tasks/fit_coadd_multiband.py

@@ -33,7 +33,7 @@
 import lsst.pipe.base as pipeBase
 import lsst.pipe.base.connectionTypes as cT
 
-import astropy
+import astropy.table
 from abc import ABC, abstractmethod
 from pydantic import Field
 from pydantic.dataclasses import dataclass
@@ -276,8 +276,7 @@ def __init__(self, initInputs, **kwargs):
         super().__init__(initInputs=initInputs, **kwargs)
         self.makeSubtask("fit_coadd_multiband")
 
-    def runQuantum(self, butlerQC, inputRefs, outputRefs):
-        inputs = butlerQC.get(inputRefs)
+    def build_catexps(self, butlerQC, inputRefs, inputs) -> list[CatalogExposureInputs]:
         id_tp = self.config.idGenerator.apply(butlerQC.quantum.dataId).catalog_id
         # This is a roundabout way of ensuring all inputs get sorted and matched
         input_refs_objs = [(getattr(inputRefs, key), inputs[key])
@@ -305,6 +304,11 @@ def runQuantum(self, butlerQC, inputRefs, outputRefs):
                 dataId=dataId, id_tract_patch=id_tp,
             )
         catexps = [catexps[band] for band in self.config.get_band_sets()[0]]
+        return catexps
+
+    def runQuantum(self, butlerQC, inputRefs, outputRefs):
+        inputs = butlerQC.get(inputRefs)
+        catexps = self.build_catexps(butlerQC, inputRefs, inputs)
         outputs = self.run(catexps=catexps, cat_ref=inputs['cat_ref'])
         butlerQC.put(outputs, outputRefs)