Fix non-finite reduced chi2 value in some situations

Herculens · Jun 3, 2024 · bed8536 · bed8536
1 parent c445f49
commit bed8536
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/herculens/LensImage/lens_image.py b/herculens/LensImage/lens_image.py
@@ -276,6 +276,9 @@ def normalized_residuals(self, data, model, mask=None):
         if mask is not None:
             # outside the mask just add pure data
             norm_res_tot += (data / noise) * (1. - mask)
+        # make sure there is no NaN or infinite values
+        norm_res_model = np.where(np.isfinite(norm_res_model), norm_res_model, 0.)
+        norm_res_tot = np.where(np.isfinite(norm_res_tot), norm_res_tot, 0.)
         return norm_res_model, norm_res_tot
 
     def reduced_chi2(self, data, model, mask=None):