Merge pull request thuml#205 from gukgurgel/gustavo

remove unused and duplicated imports
HyeongD · Jan 15, 2024 · 8f6d477 · 8f6d477
2 parents 8f7cb20 + 1227b87
commit 8f6d477
Show file tree

Hide file tree

Showing 11 changed files with 5 additions and 33 deletions.
diff --git a/data_provider/data_loader.py b/data_provider/data_loader.py
@@ -1,9 +1,6 @@
 import os
-import numpy as np
 import pandas as pd
-import os
-import torch
-from torch.utils.data import Dataset, DataLoader
+from torch.utils.data import Dataset
 from sklearn.preprocessing import StandardScaler
 from utils.timefeatures import time_features
 import warnings

diff --git a/exp/exp_basic.py b/exp/exp_basic.py
@@ -1,6 +1,5 @@
 import os
 import torch
-import numpy as np
 
 
 class Exp_Basic(object):

diff --git a/exp/exp_main.py b/exp/exp_main.py
@@ -18,7 +18,6 @@
 import time
 
 import warnings
-import matplotlib.pyplot as plt
 import numpy as np
 
 warnings.filterwarnings('ignore')

diff --git a/layers/AutoCorrelation.py b/layers/AutoCorrelation.py
@@ -1,11 +1,6 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-import matplotlib.pyplot as plt
-import numpy as np
 import math
-from math import sqrt
-import os
 
 
 class AutoCorrelation(nn.Module):

diff --git a/layers/Embed.py b/layers/Embed.py
@@ -1,7 +1,5 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from torch.nn.utils import weight_norm
 import math
 
 def compared_version(ver1, ver2):

diff --git a/layers/SelfAttention_Family.py b/layers/SelfAttention_Family.py
@@ -1,15 +1,10 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-
-import matplotlib.pyplot as plt
 
 import numpy as np
-import math
 from math import sqrt
 from utils.masking import TriangularCausalMask, ProbMask
 from reformer_pytorch import LSHSelfAttention
-import os
 
 
 class FullAttention(nn.Module):

diff --git a/layers/Transformer_EncDec.py b/layers/Transformer_EncDec.py
@@ -1,4 +1,3 @@
-import torch
 import torch.nn as nn
 import torch.nn.functional as F
 

diff --git a/models/Autoformer.py b/models/Autoformer.py
@@ -1,11 +1,8 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from layers.Embed import DataEmbedding, DataEmbedding_wo_pos
+from layers.Embed import DataEmbedding_wo_pos
 from layers.AutoCorrelation import AutoCorrelation, AutoCorrelationLayer
 from layers.Autoformer_EncDec import Encoder, Decoder, EncoderLayer, DecoderLayer, my_Layernorm, series_decomp
-import math
-import numpy as np
 
 
 class Model(nn.Module):

diff --git a/models/Informer.py b/models/Informer.py
@@ -1,11 +1,8 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from utils.masking import TriangularCausalMask, ProbMask
 from layers.Transformer_EncDec import Decoder, DecoderLayer, Encoder, EncoderLayer, ConvLayer
-from layers.SelfAttention_Family import FullAttention, ProbAttention, AttentionLayer
+from layers.SelfAttention_Family import ProbAttention, AttentionLayer
 from layers.Embed import DataEmbedding
-import numpy as np
 
 
 class Model(nn.Module):

diff --git a/models/Reformer.py b/models/Reformer.py
@@ -1,10 +1,8 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from layers.Transformer_EncDec import Decoder, DecoderLayer, Encoder, EncoderLayer, ConvLayer
+from layers.Transformer_EncDec import Encoder, EncoderLayer
 from layers.SelfAttention_Family import ReformerLayer
 from layers.Embed import DataEmbedding
-import numpy as np
 
 
 class Model(nn.Module):

diff --git a/models/Transformer.py b/models/Transformer.py
@@ -1,10 +1,8 @@
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
-from layers.Transformer_EncDec import Decoder, DecoderLayer, Encoder, EncoderLayer, ConvLayer
+from layers.Transformer_EncDec import Decoder, DecoderLayer, Encoder, EncoderLayer
 from layers.SelfAttention_Family import FullAttention, AttentionLayer
 from layers.Embed import DataEmbedding
-import numpy as np
 
 
 class Model(nn.Module):