diff --git a/src/lasp/lasp_octavefilter.py b/src/lasp/lasp_octavefilter.py index 56f1e89..fdcb39e 100644 --- a/src/lasp/lasp_octavefilter.py +++ b/src/lasp/lasp_octavefilter.py @@ -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 {}