Skip to content

Commit

Permalink
Restore missing timeMethod decorators.
Browse files Browse the repository at this point in the history
The decorators were dropped when adding Gen 3 support to certain tasks.
  • Loading branch information
kfindeisen committed Aug 10, 2020
1 parent 98646ba commit e121b9b
Show file tree
Hide file tree
Showing 6 changed files with 8 additions and 0 deletions.
3 changes: 3 additions & 0 deletions python/lsst/pipe/tasks/assembleCoadd.py
Original file line number Diff line number Diff line change
Expand Up @@ -709,6 +709,7 @@ def prepareStats(self, mask=None):
statsFlags = afwMath.stringToStatisticsProperty(self.config.statistic)
return pipeBase.Struct(ctrl=statsCtrl, flags=statsFlags)

@pipeBase.timeMethod
def run(self, skyInfo, tempExpRefList, imageScalerList, weightList,
altMaskList=None, mask=None, supplementaryData=None):
"""Assemble a coadd from input warps
Expand Down Expand Up @@ -1422,6 +1423,7 @@ def __init__(self, *args, **kwargs):
self.makeSubtask("assembleMeanCoadd")

@utils.inheritDoc(AssembleCoaddTask)
@pipeBase.timeMethod
def run(self, skyInfo, tempExpRefList, imageScalerList, weightList, *args, **kwargs):
"""Assemble the coadd for a region.
Expand Down Expand Up @@ -2075,6 +2077,7 @@ def _noTemplateMessage(self, warpType):
return message

@utils.inheritDoc(AssembleCoaddTask)
@pipeBase.timeMethod
def run(self, skyInfo, tempExpRefList, imageScalerList, weightList,
supplementaryData, *args, **kwargs):
"""Assemble the coadd.
Expand Down
1 change: 1 addition & 0 deletions python/lsst/pipe/tasks/calibrate.py
Original file line number Diff line number Diff line change
Expand Up @@ -636,6 +636,7 @@ def runQuantum(self, butlerQC, inputRefs, outputRefs):
outputs.matches = normalizedMatches
butlerQC.put(outputs, outputRefs)

@pipeBase.timeMethod
def run(self, exposure, exposureIdInfo=None, background=None,
icSourceCat=None):
"""!Calibrate an exposure (science image or coadd)
Expand Down
1 change: 1 addition & 0 deletions python/lsst/pipe/tasks/dcrAssembleCoadd.py
Original file line number Diff line number Diff line change
Expand Up @@ -565,6 +565,7 @@ def prepareDcrInputs(self, templateCoadd, warpRefList, weightList):
psf=psf)
return dcrModels

@pipeBase.timeMethod
def run(self, skyInfo, warpRefList, imageScalerList, weightList,
supplementaryData=None):
"""Assemble the coadd.
Expand Down
1 change: 1 addition & 0 deletions python/lsst/pipe/tasks/imageDifference.py
Original file line number Diff line number Diff line change
Expand Up @@ -526,6 +526,7 @@ def runDataRef(self, sensorRef, templateIdList=None):
sensorRef.put(results.subtractedExposure, subtractedExposureName)
return results

@pipeBase.timeMethod
def run(self, exposure=None, selectSources=None, templateExposure=None, templateSources=None,
idFactory=None, calexpBackgroundExposure=None, subtractedExposure=None):
"""PSF matches, subtract two images and perform detection on the difference image.
Expand Down
1 change: 1 addition & 0 deletions python/lsst/pipe/tasks/makeCoaddTempExp.py
Original file line number Diff line number Diff line change
Expand Up @@ -379,6 +379,7 @@ def runDataRef(self, patchRef, selectDataList=[]):

return dataRefList

@pipeBase.timeMethod
def run(self, calExpList, ccdIdList, skyInfo, visitId=0, dataIdList=None):
"""Create a Warp from inputs
Expand Down
1 change: 1 addition & 0 deletions python/lsst/pipe/tasks/transformMeasurement.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ def getSchemaCatalogs(self):
transformedSrc = afwTable.BaseCatalog(self.mapper.getOutputSchema())
return {self.outputDataset: transformedSrc}

@pipeBase.timeMethod
def run(self, inputCat, wcs, photoCalib):
"""!Transform raw source measurements to calibrated quantities.
Expand Down

0 comments on commit e121b9b

Please sign in to comment.