lint

Author: duyifanict

.pylintrc

@@ -12,7 +12,7 @@ ignore=baselines,assets,checkpoints,examples,scripts
 
 # Files or directories matching the regex patterns are skipped. The regex
 # matches against base names, not paths.
-ignore-patterns=^\.|^_|^.*\.md|^.*\.txt|^.*\.csv|^.*\.CFF|^LICENSE
+ignore-patterns=^\.|^_|^.*\.md|^.*\.txt|^.*\.csv|^.*\.CFF|^LICENSE|^.*\.toml
 
 # Pickle collected data for later comparisons.
 persistent=no

src/basicts/configs/base_config.py

@@ -9,19 +9,21 @@
 from functools import partial
 from numbers import Number
 from types import FunctionType
-from typing import Callable, List, Literal, Optional, Tuple, TYPE_CHECKING, Union
+from typing import (TYPE_CHECKING, Callable, List, Literal, Optional, Tuple,
+                    Union)
 
 import numpy as np
 import torch
 from easydict import EasyDict
 from torch.optim.lr_scheduler import LRScheduler
 
+from .model_config import BasicTSModelConfig
+
 # avoid circular imports
 if TYPE_CHECKING:
     from basicts.runners.callback import BasicTSCallback
     from basicts.runners.taskflow import BasicTSTaskFlow
 
-from .model_config import BasicTSModelConfig
 
 
 @dataclass(init=False)

src/basicts/data/blast.py

@@ -244,12 +244,12 @@ def __getitem__(self, idx: int) -> tuple:
 
     def __len__(self):
         return self.data.shape[0]
-    
+
     def __getstate__(self):
         state = self.__dict__.copy()
-        del state['_data']
+        del state["_data"]
         return state
-    
+
     def __setstate__(self, state):
         self.__dict__.update(state)
         self._data = self._load_data()

src/basicts/models/DUET/arch/mahalanobis_mask.py

@@ -1,3 +1,4 @@
+# pylint: disable=not-callable
 import torch
 import torch.nn.functional as F
 from einops import rearrange

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

@@ -1,3 +1,4 @@
+# pylint: disable=not-callable
 from typing import Callable, Dict
 
 import torch

src/basicts/models/FiLM/arch/film_arch.py

@@ -1,3 +1,4 @@
+# pylint: disable=not-callable
 import numpy as np
 import torch
 import torch.nn.functional as F

src/basicts/models/FiLM/config/film_config.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass, field
-from typing import Sequence
 
 from basicts.configs import BasicTSModelConfig
 

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

@@ -1,3 +1,4 @@
+# pylint: disable=not-callable
 import torch
 from torch import nn
 from torch.nn import functional as F

src/basicts/models/Koopa/arch/layers.py

@@ -1,3 +1,4 @@
+# pylint: disable=not-callable
 import math
 
 import torch

src/basicts/models/Koopa/config/koopa_config.py

@@ -17,4 +17,4 @@ class KoopaConfig(BasicTSModelConfig):
     dynamic_dim: int = field(default=64, metadata={"help": "Dynamic feature dimension. Must be > 0."})
     hidden_dim: int = field(default=64, metadata={"help": "Hidden dimension."})
     hidden_layers: int = field(default=2, metadata={"help": "Number of hidden layers (>=2 recommended)."})
-    multistep: bool = field(default=False, metadata={"help": "Whether to use multistep forecasting."})
+    multistep: bool = field(default=False, metadata={"help": "Whether to use multistep forecasting."})