Make input optional for get_detection_mask

lsst-dm · Oct 3, 2024 · b8afcdb · b8afcdb
1 parent 4d97e61
commit b8afcdb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/metadetect/lsst/util.py b/metadetect/lsst/util.py
@@ -825,7 +825,7 @@ def get_stats_mask(exp):
     return stats_mask
 
 
-def get_detection_mask(exp):
+def get_detection_mask(exp=None):
     """
     Get a mask for detection. Regions with these flags set will not be searched
     for objects.
@@ -838,7 +838,7 @@ def get_detection_mask(exp):
 
     Parameters
     ----------
-    exp: lsst.afw.image.ExposureF
+    exp: lsst.afw.image.ExposureF, optional
         The exposure
 
     Returns
@@ -848,7 +848,7 @@ def get_detection_mask(exp):
 
     mask = ['EDGE', 'NO_DATA']
 
-    if 'BRIGHT' in exp.mask.getMaskPlaneDict():
+    if exp is not None and 'BRIGHT' in exp.mask.getMaskPlaneDict():
         mask += ['BRIGHT']
 
     return mask