small fixes

Author: ayasyrev

.flake8

@@ -2,6 +2,9 @@
 select = C,E,F,W
 max-complexity = 10
 max-line-length = 120
-
-application-import-names = model_constructor
-import-order-style = google
+disable-noqa = True
+application-import-names = model_constructor, tests
+import-order-style = google
+per-file-ignores =
+    # imported but unused
+    __init__.py: F401

src/model_constructor/__init__.py

@@ -1,3 +1,3 @@
-from .convmixer import ConvMixer  # noqa F401
-from .model_constructor import ModelConstructor, ModelCfg  # noqa F401
-from .version import __version__  # noqa F401
+from .convmixer import ConvMixer
+from .model_constructor import ModelConstructor, ModelCfg
+from .version import __version__

src/model_constructor/helpers.py

@@ -11,7 +11,7 @@
 
 def nn_seq(list_of_tuples: Iterable[tuple[str, nn.Module]]) -> nn.Sequential:
     """return nn.Sequential from OrderedDict from list of tuples"""
-    return nn.Sequential(OrderedDict(list_of_tuples))  #
+    return nn.Sequential(OrderedDict(list_of_tuples))
 
 
 def init_cnn(module: nn.Module) -> None:
@@ -53,7 +53,7 @@ def __repr_changed_args__(self) -> list[str]:
         """Return list repr for changed fields"""
         return [
             f"{field}: {self._get_str_value(field)}"
-            for field in self.model_fields_set
+            for field in self.model_fields_set  # pylint: disable=E1133
             if field != "name"
         ]
 

src/model_constructor/model_constructor.py

@@ -190,7 +190,7 @@ def create_model(
     ) -> nn.Sequential:
         if cfg:
             return cls(**cfg.model_dump())()
-        return cls(**kwargs)()
+        return cls(**kwargs)()  # type: ignore
 
     def __call__(self) -> nn.Sequential:
         """Create model."""

tests/test_mc.py

@@ -3,8 +3,7 @@
 import torch
 
 from model_constructor.blocks import BottleneckBlock
-from model_constructor.layers import (SEModule, SEModuleConv,
-                                      SimpleSelfAttention)
+from model_constructor.layers import SEModule, SEModuleConv, SimpleSelfAttention
 from model_constructor.model_constructor import ModelConstructor
 
 bs_test = 4