Fixes downsample error MIT-LCP#249

Lucas-Mc · Lucas-Mc · commit 3fd3cc70d5d6 · 2020-08-20T12:19:36.000-04:00
Fixes error when downsampling single channel if NaN values are present. This was a Scipy issue so extra pre-processing and input validation was done to help safe-guard against this issue re-occuring in the future. Fixes MIT-LCP#249.
diff --git a/wfdb/processing/basic.py b/wfdb/processing/basic.py
@@ -1,5 +1,6 @@
 import numpy as np
 from scipy import signal
+import pandas as pd
 import pdb
 
 from wfdb.io.annotation import Annotation
@@ -91,6 +92,9 @@ def resample_sig(x, fs, fs_target):
         return x, t
 
     new_length = int(x.shape[0]*fs_target/fs)
+    # Resample the array if NaN values are present
+    if np.isnan(x).any():
+        x = pd.Series(x.reshape((-1,))).interpolate().values
     resampled_x, resampled_t = signal.resample(x, num=new_length, t=t)
     assert resampled_x.shape == resampled_t.shape and resampled_x.shape[0] == new_length
     assert np.all(np.diff(resampled_t) > 0)
