update

Author: LinglongQian

pypots/imputation/fedformer/model.py

@@ -111,13 +111,13 @@ class FEDformer(BaseNNImputer):
 
     def __init__(
         self,
-        n_steps,
-        n_features,
-        n_layers,
-        d_model,
-        n_heads,
-        d_ffn,
-        moving_avg_window_size,
+        n_steps: int,
+        n_features: int,
+        n_layers: int,
+        d_model: int,
+        n_heads: int,
+        d_ffn: int,
+        moving_avg_window_size: int,
         dropout: float = 0,
         version="Fourier",
         modes=32,