fix: shape mismatch in features.rms

ifsm · Jun 25, 2024 · 7d07dc8 · 7d07dc8
1 parent fb39fea
commit 7d07dc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/apollon/signal/features.py b/src/apollon/signal/features.py
@@ -147,7 +147,7 @@ def rms(sig: FloatArray) -> FloatArray:
         RMS of signal along first axis.
     """
     buff = _np.empty_like(sig, dtype=_np.double)
-    out = _np.empty((sig.shape[1], 1), dtype=_np.double)
+    out = _np.empty((1, sig.shape[1]), dtype=_np.double)
     _np.abs(sig, out=buff)
     _np.square(buff, out=buff)
     _np.mean(buff, axis=0, keepdims=True, out=out)