Fixing mask issues

lsst · Oct 17, 2023 · e4e8c90 · e4e8c90
1 parent 7d888a3
commit e4e8c90
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/python/lsst/ip/diffim/subtractImages.py b/python/lsst/ip/diffim/subtractImages.py
@@ -1087,8 +1087,12 @@ def _interpolateImage(maskedImage, badMaskPlanes, fallbackValue=None):
     result: `float`
         The number of masked pixels that were replaced.
     """
+    imgBadMaskPlanes = [
+            maskPlane for maskPlane in badMaskPlanes if maskPlane in maskedImage.mask.getMaskPlaneDict()
+        ]
+
     image = maskedImage.image.array
-    badPixels = (maskedImage.mask.array & maskedImage.mask.getPlaneBitMask(badMaskPlanes)) > 0
+    badPixels = (maskedImage.mask.array & maskedImage.mask.getPlaneBitMask(imgBadMaskPlanes)) > 0
     image[badPixels] = np.nan
     if fallbackValue is None:
         fallbackValue = np.nanmedian(image)

diff --git a/tests/test_subtractTask.py b/tests/test_subtractTask.py
@@ -757,7 +757,6 @@ def test_fake_mask_plane_propagation(self):
         template_fake_masked = (template.mask.array & template.mask.getPlaneBitMask("FAKE")) > 0
 
         config = subtractImages.AlardLuptonSubtractTask.ConfigClass()
-        config.mode = "convolveTemplate"
         task = subtractImages.AlardLuptonSubtractTask(config=config)
         subtraction = task.run(template, science, sources)