Fix for MIT-LCP/wfdb-python issue MIT-LCP#269. The number of subplots (n_subplots) should match the number of ylabels regardless of whether records or annotations are being passed in.

briangow · briangow · commit dcedbc469151 · 2020-12-22T14:08:03.000-05:00
diff --git a/wfdb/plot/plot.py b/wfdb/plot/plot.py
@@ -128,9 +128,9 @@ def plot_items(signal=None, ann_samp=None, ann_sym=None, fs=None,
                             'signal: {} values'.format(signal.shape[1]))
 
     if ylabel:
-        if len(ylabel) != signal.shape[1]:
+        if len(ylabel) != n_subplots:
             raise Exception('The length of the ylabel must be the same as the '
-                            'signal: {} values'.format(signal.shape[1]))
+                            'signal: {} values'.format(n_subplots))
 
     label_figure(axes, n_subplots, time_units, sig_name, sig_units,
                  xlabel, ylabel, title)
