fix lint

Author: duyifanict

src/basicts/configs/base_config.py

@@ -13,11 +13,10 @@
 
 import numpy as np
 import torch
-from easydict import EasyDict
-from torch.optim.lr_scheduler import LRScheduler
-
 from basicts.runners.callback import BasicTSCallback
 from basicts.runners.taskflow import BasicTSTaskFlow
+from easydict import EasyDict
+from torch.optim.lr_scheduler import LRScheduler
 
 from .model_config import BasicTSModelConfig
 

src/basicts/configs/tsc_config.py

@@ -2,13 +2,12 @@
 from typing import Callable, List, Literal, Tuple, Union
 
 import numpy as np
-from torch.nn import CrossEntropyLoss
-from torch.optim import Adam
-
 from basicts.data import UEADataset
 from basicts.runners.callback import BasicTSCallback
 from basicts.runners.taskflow import (BasicTSClassificationTaskFlow,
                                       BasicTSTaskFlow)
+from torch.nn import CrossEntropyLoss
+from torch.optim import Adam
 
 from .base_config import BasicTSConfig
 from .model_config import BasicTSModelConfig

src/basicts/configs/tsf_config.py

@@ -2,13 +2,12 @@
 from typing import Callable, List, Literal, Tuple, Union
 
 import numpy as np
-from torch.optim import Adam
-
 from basicts.data import BasicTSForecastingDataset
 from basicts.runners.callback import BasicTSCallback
 from basicts.runners.taskflow import (BasicTSForecastingTaskFlow,
                                       BasicTSTaskFlow)
 from basicts.scaler import ZScoreScaler
+from torch.optim import Adam
 
 from .base_config import BasicTSConfig
 from .model_config import BasicTSModelConfig

src/basicts/configs/tsfm_config.py

@@ -3,13 +3,12 @@
 
 import numpy as np
 import torch
-from torch.optim import AdamW
-
 from basicts.data import BasicTSForecastingDataset
 from basicts.runners.callback import BasicTSCallback
+from basicts.runners.optim.lr_schedulers import CosineWarmup
 from basicts.runners.taskflow import (BasicTSForecastingTaskFlow,
                                       BasicTSTaskFlow)
-from basicts.runners.optim.lr_schedulers import CosineWarmup
+from torch.optim import AdamW
 
 from .base_config import BasicTSConfig
 from .model_config import BasicTSModelConfig

src/basicts/configs/tsi_config.py

@@ -2,12 +2,11 @@
 from typing import Callable, List, Literal, Tuple, Union
 
 import numpy as np
-from torch.optim import Adam
-
 from basicts.data import BasicTSImputationDataset
 from basicts.runners.callback import BasicTSCallback
 from basicts.runners.taskflow import BasicTSImputationTaskFlow, BasicTSTaskFlow
 from basicts.scaler import ZScoreScaler
+from torch.optim import Adam
 
 from .base_config import BasicTSConfig
 from .model_config import BasicTSModelConfig

src/basicts/models/FITS/arch/fits_arch.py

@@ -1,10 +1,9 @@
 from typing import Callable, Dict
 
 import torch
-from torch import nn
-
 from basicts.metrics import ALL_METRICS
 from basicts.modules.norm import RevIN
+from torch import nn
 
 from ..config.fits_config import FITSConfig
 

src/basicts/models/FreTS/arch/frets_arch.py

@@ -1,9 +1,8 @@
 import torch
+from basicts.modules import MLPLayer
 from torch import nn
 from torch.nn import functional as F
 
-from basicts.modules import MLPLayer
-
 from ..config.frets_config import FreTSConfig
 
 

src/basicts/models/Leddam/arch/leddam_arch.py

@@ -1,10 +1,9 @@
 import torch
-from torch import nn
-
 from basicts.modules import MLPLayer
 from basicts.modules.embed import PositionEmbedding, SequenceEmbedding
 from basicts.modules.norm import RevIN
 from basicts.modules.transformer import Encoder, MultiHeadAttention
+from torch import nn
 
 from ..config.leddam_config import LeddamConfig
 from .leddam_layers import (AutoAttention, LearnableDecomposition,

src/basicts/models/Leddam/arch/leddam_layers.py

@@ -3,9 +3,8 @@
 
 import torch
 import torch.nn.functional as F
-from torch import nn
-
 from basicts.modules.transformer import EncoderLayer
+from torch import nn
 
 
 class LeddamEncoderLayer(EncoderLayer):