From aae57853f685e91375fc4fc6fff8d7fd906123c7 Mon Sep 17 00:00:00 2001 From: Brianna Smart Date: Fri, 19 Apr 2024 14:15:20 -0700 Subject: [PATCH] Remove longTrailedSources from pipeline output --- python/lsst/ap/verify/testPipeline.py | 5 ++--- tests/test_testPipeline.py | 4 ++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/python/lsst/ap/verify/testPipeline.py b/python/lsst/ap/verify/testPipeline.py index 02d0b646..a0bade07 100644 --- a/python/lsst/ap/verify/testPipeline.py +++ b/python/lsst/ap/verify/testPipeline.py @@ -457,6 +457,7 @@ def run(self, diaSourceCat): """ return Struct(filteredDiaSourceCat=afwTable.SourceCatalog(), rejectedDiaSources=afwTable.SourceCatalog(), + longTrailedSources=afwTable.SourceCatalog(), ) @@ -631,6 +632,4 @@ def run(self, return Struct(apdbMarker=self.config.apdb.value, associatedDiaSources=pandas.DataFrame(), diaForcedSources=pandas.DataFrame(), - diaObjects=pandas.DataFrame(), - longTrailedSources=pandas.DataFrame(), - ) + diaObjects=pandas.DataFrame(),) diff --git a/tests/test_testPipeline.py b/tests/test_testPipeline.py index 290b94cc..be4f3f78 100644 --- a/tests/test_testPipeline.py +++ b/tests/test_testPipeline.py @@ -131,7 +131,8 @@ def setUpClass(cls): butlerTests.addDatasetType(cls.repo, "visitSsObjects", cls.visitOnlyId.dimensions, "DataFrame") butlerTests.addDatasetType(cls.repo, "apdb_marker", cls.visitId.dimensions, "Config") butlerTests.addDatasetType(cls.repo, "deepDiff_assocDiaSrc", cls.visitId.dimensions, "DataFrame") - butlerTests.addDatasetType(cls.repo, "deepDiff_longTrailedSrc", cls.visitId.dimensions, "DataFrame") + butlerTests.addDatasetType(cls.repo, "deepDiff_longTrailedSrc", cls.visitId.dimensions, + "ArrowAstropy") butlerTests.addDatasetType(cls.repo, "deepRealBogusSources", cls.visitId.dimensions, "Catalog") butlerTests.addDatasetType(cls.repo, "deepDiff_diaForcedSrc", cls.visitId.dimensions, "DataFrame") butlerTests.addDatasetType(cls.repo, "deepDiff_diaObject", cls.visitId.dimensions, "DataFrame") @@ -358,7 +359,6 @@ def testMockDiaPipelineTask(self): "template": self.visitId, "apdbMarker": self.visitId, "associatedDiaSources": self.visitId, - "longTrailedSources": self.visitId, "diaForcedSources": self.visitId, "diaObjects": self.visitId, })