Fixed assert in octave filter

This commit is contained in:
Thijs Hekman 2022-10-11 10:50:45 +02:00
parent 6aa1262c73
commit 6ed88af3e2
1 changed files with 4 additions and 2 deletions

View File

@ -33,8 +33,10 @@ class OverallFilterBank:
"""
Filter input data
"""
assert data.ndim == 2
assert data.shape[1] == 1, "invalid number of channels, should be 1"
if data.ndim == 1:
data = data[:, None]
elif data.ndim == 2:
assert data.shape[1] == 1, "invalid number of channels, should be 1"
if data.shape[0] == 0:
return {}