handle the case for SourceNotVisible exception (#386)

Co-authored-by: Ismael Mendoza <[email protected]>
LSSTDESC · Mar 8, 2023 · cb58019 · cb58019
1 parent 2e062c9
commit cb58019
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"):