Skip to content

Commit

Permalink
Match new TransformDiaSourceCatalog API
Browse files Browse the repository at this point in the history
  • Loading branch information
parejkoj committed May 1, 2024
1 parent c735040 commit 22ce28b
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
10 changes: 5 additions & 5 deletions python/lsst/ap/verify/testPipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -514,15 +514,15 @@ def __init__(self, initInputs, **kwargs):

def runQuantum(self, butlerQC, inputRefs, outputRefs):
inputs = butlerQC.get(inputRefs)
idGenerator = self.config.idGenerator.apply(butlerQC.quantum.dataId)
inputs["ccdVisitId"] = idGenerator.catalog_id
inputs["band"] = butlerQC.quantum.dataId["band"]
inputs["visit"] = butlerQC.quantum.dataId["visit"]
inputs["detector"] = butlerQC.quantum.dataId["detector"]

outputs = self.run(**inputs)

butlerQC.put(outputs, outputRefs)

def run(self, diaSourceCat, diffIm, band, ccdVisitId, funcs=None):
def run(self, diaSourceCat, diffIm, band, visit, detector, funcs=None):
"""Produce transformation outputs with no processing.
Parameters
Expand All @@ -533,8 +533,8 @@ def run(self, diaSourceCat, diffIm, band, ccdVisitId, funcs=None):
An image, to provide supplementary information.
band : `str`
The band in which the sources were observed.
ccdVisitId : `int`
The exposure ID in which the sources were observed.
visit, detector: `int`
Visit and detector the sources were detected on.
funcs, optional
Unused.
Expand Down
2 changes: 1 addition & 1 deletion tests/test_testPipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,7 @@ def testMockRBTransiNetTask(self):
def testMockTransformDiaSourceCatalogTask(self):
task = MockTransformDiaSourceCatalogTask(initInputs=afwTable.SourceCatalog())
pipelineTests.assertValidInitOutput(task)
result = task.run(afwTable.SourceCatalog(), afwImage.ExposureF(), 'k', 42)
result = task.run(afwTable.SourceCatalog(), afwImage.ExposureF(), 'k', 42, 2)
pipelineTests.assertValidOutput(task, result)

self.butler.put(afwTable.SourceCatalog(), "deepDiff_candidateDiaSrc", self.visitId)
Expand Down

0 comments on commit 22ce28b

Please sign in to comment.