Skip to content

Commit d9f04c5

Browse files
committed
fix res estimator
1 parent 4a22256 commit d9f04c5

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

wfdb/io/_signal.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1506,17 +1506,20 @@ def est_res(signals):
15061506
# Estimate the number of steps as the range divided by the
15071507
# minimum increment.
15081508
if isinstance(signals, list):
1509-
sortedsig = np.sort(signals[ch])
1509+
sorted_sig = np.sort(np.unique(signals[ch]))
15101510
else:
1511-
sortedsig = np.sort(signals[:,ch])
1511+
if signals.ndim == 1:
1512+
sorted_sig = np.sort(np.unique(signals))
1513+
else:
1514+
sorted_sig = np.sort(np.unique(signals[:,ch]))
15121515

1513-
min_inc = min(np.diff(sortedsig))
1516+
min_inc = min(np.diff(sorted_sig))
15141517

15151518
if min_inc == 0:
15161519
# Case where signal is flat. Resolution is 0.
15171520
res.append(0)
15181521
else:
1519-
nlevels = 1 + (sortedsig[-1]-sortedsig[0])/min_inc
1522+
nlevels = 1 + (sorted_sig[-1]-sorted_sig[0]) / min_inc
15201523
if nlevels >= res_levels[-1]:
15211524
res.append(32)
15221525
else:

0 commit comments

Comments
 (0)