Merge branch 'dev' into (feat)add_informer

WenjieDu · Apr 1, 2024 · 708f9fe · 708f9fe
2 parents a1e127a + 46e2434
commit 708f9fe
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pypots/imputation/autoformer/modules/core.py b/pypots/imputation/autoformer/modules/core.py
@@ -8,7 +8,6 @@
 import torch.nn as nn
 
 from .submodules import (
-    SeriesDecompositionBlock,
     SeasonalLayerNorm,
     AutoformerEncoderLayer,
     AutoformerEncoder,
@@ -38,7 +37,6 @@ def __init__(
 
         self.seq_len = n_steps
         self.n_layers = n_layers
-        self.series_decomp = SeriesDecompositionBlock(moving_avg_window_size)
         self.enc_embedding = DataEmbedding(
             n_features,
             d_model,
@@ -59,7 +57,7 @@ def __init__(
                     dropout,
                     activation,
                 )
-                for i in range(n_layers)
+                for _ in range(n_layers)
             ],
             norm_layer=SeasonalLayerNorm(d_model),
         )