handle all nans - wfdb format benchmarking

chorus-ai · May 10, 2024 · 674d9c4 · 674d9c4
1 parent ac7cade
commit 674d9c4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/waveform_benchmark/formats/wfdb.py b/waveform_benchmark/formats/wfdb.py
@@ -43,9 +43,12 @@ def write_waveforms(self, path, waveforms):
                 sig_samples[start:end] = chunk['samples']
                 sig_gain = max(sig_gain, chunk['gain'])
 
-            sample_min = numpy.nanmin(sig_samples)
-            sample_max = numpy.nanmax(sig_samples)
-            sig_baseline = round(-sig_gain * (sample_min + sample_max) / 2)
+            if all(numpy.isnan(sig_samples)):
+                sig_baseline = 0
+            else:
+                sample_min = numpy.nanmin(sig_samples)
+                sample_max = numpy.nanmax(sig_samples)
+                sig_baseline = round(-sig_gain * (sample_min + sample_max) / 2)
             adc_gain.append(sig_gain)
             baseline.append(sig_baseline)
             e_p_signal.append(sig_samples)