]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
refactor(qav3): add missing type
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Wed, 14 May 2025 17:37:06 +0000 (19:37 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Wed, 14 May 2025 17:37:06 +0000 (19:37 +0200)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
quickadapter/user_data/strategies/Utils.py

index 6b5554911d1dbb4b9c93a9c9b4b9c15852e4ff19..087ffda0a8caa2113a77f0cb86c729ac231bbe31 100644 (file)
@@ -43,14 +43,16 @@ def derive_gaussian_std_from_window(window: int) -> float:
     return (window - 1) / 6.0 if window > 1 else 0.5
 
 
-def zero_phase_gaussian(series: pd.Series, window: int, std: float):
+def zero_phase_gaussian(series: pd.Series, window: int, std: float) -> pd.Series:
     kernel = gaussian(window, std=std)
     kernel /= kernel.sum()
 
     padding_length = window - 1
-    padded_series = np.pad(series.values, (padding_length, padding_length), mode="edge")
+    padded_series_values = np.pad(
+        series.values, (padding_length, padding_length), mode="edge"
+    )
 
-    forward = convolve(padded_series, kernel, mode="valid")
+    forward = convolve(padded_series_values, kernel, mode="valid")
     backward = convolve(forward[::-1], kernel, mode="valid")[::-1]
 
     return pd.Series(backward, index=series.index)