Explicitly handle NaNs in metric evaluation

Author: MaxvandenBoom

erdetect/core/metrics/metric_cross_proj.py

@@ -85,6 +85,8 @@ def evaluate_callback(channel_index, stimpair_index, metric_values):
         # (double) The threshold which needs to be exceeded to detect a peak
         cross_proj_threshold = config('detection', 'cross_proj', 'threshold')
 
+        if metric_values[channel_index, stimpair_index, 0] == np.nan:
+            return False
         return metric_values[channel_index, stimpair_index, 0] > cross_proj_threshold
 
     @staticmethod

erdetect/core/metrics/metric_waveform.py

@@ -98,6 +98,9 @@ def evaluate_callback(channel_index, stimpair_index, metric_values):
         # (double) The threshold which needs to be exceeded to detect a peak
         waveform_threshold = config('detection', 'waveform', 'threshold')
 
+        # evaluate
+        if metric_values[channel_index, stimpair_index] == np.nan:
+            return False
         return metric_values[channel_index, stimpair_index] > waveform_threshold
 
     @staticmethod