Handle NaN in FlatField class

nabu/preproc/flatfield.py

@@ -190,6 +190,12 @@ class FlatFieldArrays:
             self._dark = dark
         return self._dark
 
+    def remove_invalid_values(self, img):
+        if self.nan_value is None:
+            return
+        invalid_mask = np.logical_not(np.isfinite(img))
+        img[invalid_mask] = self.nan_value
+
     def normalize_radios(self, radios):
         """
         Apply a flat-field normalization, with the current parameters, to a stack
@@ -211,6 +217,7 @@ class FlatFieldArrays:
             if do_flats_distortion_correction:
                 flat = self.distortion_correction.estimate_and_correct(flat, radio_data)
             radios[i] = radio_data / flat
+            self.remove_invalid_values(radios[i])
         return radios
 
 
@@ -225,6 +232,7 @@ class FlatFieldArrays:
         if self.distortion_correction is not None:
             flat = self.distortion_correction.estimate_and_correct(flat, radio)
         radio /= flat
+        self.remove_invalid_values(radio)
         return radio