Only add the invalid psf flag if it is not present in the schema.

lsst · May 7, 2024 · 743cd87 · 743cd87
1 parent 3beeb9b
commit 743cd87
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/python/lsst/meas/base/baseMeasurement.py b/python/lsst/meas/base/baseMeasurement.py
@@ -318,11 +318,15 @@ def initializePlugins(self, **kwds):
         else:
             schema = kwds["schema"]
 
-        self.keyInvalidPsf = schema.addField(
-            "base_InvalidPsf_flag",
-            type="Flag",
-            doc="Invalid PSF at this location.",
-        )
+        invalidPsfName = "base_InvalidPsf_flag"
+        if invalidPsfName in schema:
+            self.keyInvalidPsf = schema.find(invalidPsfName).key
+        else:
+            self.keyInvalidPsf = schema.addField(
+                invalidPsfName,
+                type="Flag",
+                doc="Invalid PSF at this location.",
+            )
 
     def callMeasure(self, measRecord, *args, **kwds):
         """Call ``measure`` on all plugins and consistently handle exceptions.