Use updated getChildren in references

lsst · May 11, 2021 · 9f9d2ef · 9f9d2ef
1 parent ad42ec4
commit 9f9d2ef
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/lsst/meas/base/references.py b/python/lsst/meas/base/references.py
@@ -202,11 +202,11 @@ def subset(self, sources, bbox, wcs):
         parentSources = catalog.getChildren(0)
         skyCoordList = [source.getCoord() for source in parentSources]
         pixelPosList = wcs.skyToPixel(skyCoordList)
-        for parent, pixel in zip(parentSources, pixelPosList):
-            if boxD.contains(pixel):
-                yield parent
-                for child in catalog.getChildren(parent.getId()):
-                    yield child
+        parentList = [parent for parent, pixel in zip(parentSources, pixelPosList) if boxD.contains(pixel)]
+        childrenIter = catalog.getChildren((parent.getId() for parent in parentList))
+        for parent, children in zip(parentList, childrenIter):
+            yield parent
+            yield from children
 
 
 class CoaddSrcReferencesConfig(BaseReferencesTask.ConfigClass):