Fixed assert in octave filter

2022-10-11 10:50:45 +02:00 · 2022-10-11 10:50:45 +02:00 · 6ed88af3e2
commit 6ed88af3e2
parent 6aa1262c73
1 changed files with 4 additions and 2 deletions
--- a/src/lasp/lasp_octavefilter.py
+++ b/src/lasp/lasp_octavefilter.py
@ -33,7 +33,9 @@ class OverallFilterBank:
        """
        Filter input data
        """
-        assert data.ndim == 2
+        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: