more error catching

LSSTDESC · Apr 17, 2024 · 5787596 · 5787596
1 parent 7aac46f
commit 5787596
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/btk/deblend.py b/btk/deblend.py
@@ -396,6 +396,12 @@ def deblend(self, ii: int, blend_batch: BlendBatch) -> DeblendExample:
                 band_image, self.thresh, err=bkg.globalrms, segmentation_map=False
             )
 
+            if len(catalog) > self.max_n_sources:
+                raise ValueError(
+                    "SEP predicted more sources than `max_n_sources`. Consider increasing `thresh`"
+                    " or `max_n_sources`."
+                )
+
             # convert predictions to arcseconds
             ra_detections, dec_detections = wcs.pixel_to_world_values(catalog["x"], catalog["y"])
             ra_detections *= 3600