From 9e7a9507c6aacc3eccc725eb7509828bbf43048c Mon Sep 17 00:00:00 2001
From: Wenjie Du <wenjay.du@gmail.com>
Date: Fri, 29 Mar 2024 23:14:21 +0800
Subject: [PATCH] refactor: remove useless module in Autoformer;

---
 pypots/imputation/autoformer/modules/core.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/pypots/imputation/autoformer/modules/core.py b/pypots/imputation/autoformer/modules/core.py
index 47a2a51b..40837f6f 100644
--- a/pypots/imputation/autoformer/modules/core.py
+++ b/pypots/imputation/autoformer/modules/core.py
@@ -7,16 +7,15 @@
 
 import torch.nn as nn
 
-from ....utils.metrics import calc_mse
 from .submodules import (
     DataEmbedding_wo_Pos,
-    SeriesDecompositionBlock,
     SeasonalLayerNorm,
     AutoformerEncoderLayer,
     AutoformerEncoder,
     AutoCorrelation,
     AutoCorrelationLayer,
 )
+from ....utils.metrics import calc_mse
 
 
 class _Autoformer(nn.Module):
@@ -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_wo_Pos(
             n_features,
             d_model,
@@ -58,7 +56,7 @@ def __init__(
                     dropout,
                     activation,
                 )
-                for i in range(n_layers)
+                for _ in range(n_layers)
             ],
             norm_layer=SeasonalLayerNorm(d_model),
         )