handling NaNs in cross correlation

sdss · Sep 24, 2024 · 374a30a · 374a30a
1 parent 00e8f6c
commit 374a30a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/lvmdrp/core/spectrum1d.py b/python/lvmdrp/core/spectrum1d.py
@@ -319,6 +319,7 @@ def _cross_match_float(
         # Normalize the cross correlation
         cross_corr = cross_corr.astype(numpy.float32)
         cross_corr /= norm(stretched_signal1) * norm(obs_spec_)
+        cross_corr = numpy.nan_to_num(cross_corr)
 
         # Get the correlation shifts
         shifts = signal.correlation_lags(