handle the case for SourceNotVisible exception (#386)

Co-authored-by: Ismael Mendoza <[email protected]>
LSSTDESC · Nov 2, 2022 · c36b185 · c36b185
1 parent fd2f1aa
commit c36b185
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/btk/draw_blends.py b/btk/draw_blends.py
@@ -437,8 +437,11 @@ def render_blend(self, blend_catalog, psf, filt, survey, seedseq_blend, extra_da
 
         for k, entry in enumerate(blend_catalog):
             single_image = self.render_single(entry, filt, psf, survey, extra_data[k])
-            iso_image[k] = single_image.array
-            _blend_image += single_image
+            if single_image is None:
+                iso_image[k] = np.zeros(single_image)
+            else:
+                iso_image[k] = single_image.array
+                _blend_image += single_image
 
         # add noise.
         if self.add_noise in ("galaxy", "all"):