Add GetDcrTemplateTask

lsst · Mar 25, 2022 · af55a3d · af55a3d
1 parent 74a3074
commit af55a3d
Show file tree

Hide file tree

Showing 2 changed files with 242 additions and 83 deletions.
diff --git a/python/lsst/ip/diffim/dcrModel.py b/python/lsst/ip/diffim/dcrModel.py
@@ -118,7 +118,7 @@ def fromImage(cls, maskedImage, dcrNumSubfilters, effectiveWavelength, bandwidth
                    mask=mask, variance=variance, photoCalib=photoCalib)
 
     @classmethod
-    def fromQuantum(cls, availableCoaddRefs, effectiveWavelength, bandwidth):
+    def fromQuantum(cls, availableCoaddRefs, effectiveWavelength, bandwidth, numSubfilters):
         """Load an existing DcrModel from a Gen 3 repository.
 
         Parameters
@@ -130,6 +130,8 @@ def fromQuantum(cls, availableCoaddRefs, effectiveWavelength, bandwidth):
             The effective wavelengths of the current filter, in nanometers.
         bandwidth : `float`
             The bandwidth of the current filter, in nanometers.
+        numSubfilters : `int`
+            Number of subfilters in the DcrCoadd.
 
         Returns
         -------
@@ -143,7 +145,7 @@ def fromQuantum(cls, availableCoaddRefs, effectiveWavelength, bandwidth):
         mask = None
         variance = None
         photoCalib = None
-        modelImages = [None]*len(availableCoaddRefs)
+        modelImages = [None]*numSubfilters
 
         for coaddRef in availableCoaddRefs:
             subfilter = coaddRef.dataId["subfilter"]