Update ema_filter.py

This commit is contained in:
Shreya 2024-10-31 01:33:49 +05:30 committed by GitHub
parent 47a6069968
commit 06775c4bf5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -31,7 +31,7 @@ class EMAFilter:
if not (0 < alpha <= 1): if not (0 < alpha <= 1):
raise ValueError("Alpha must be between 0 and 1.") raise ValueError("Alpha must be between 0 and 1.")
self.alpha = alpha self.alpha = alpha
self.ema_value = None self.ema_value = 0.0
def apply(self, audio_signal: list[float]) -> np.ndarray: def apply(self, audio_signal: list[float]) -> np.ndarray:
""" """
@ -51,7 +51,7 @@ class EMAFilter:
... [0.1, 0.18, 0.304, 0.3632, 0.35056, 0.460448, 0.4483584]) ... [0.1, 0.18, 0.304, 0.3632, 0.35056, 0.460448, 0.4483584])
True True
""" """
ema_signal = [] ema_signal: list[float] = []
for sample in audio_signal: for sample in audio_signal:
if self.ema_value is None: if self.ema_value is None:
# Initialize the EMA with the first sample # Initialize the EMA with the first sample