isort src

Author: ayasyrev

src/model_constructor/__init__.py

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

src/model_constructor/activations.py

@@ -1,9 +1,8 @@
 # forked from https://github.com/rwightman/pytorch-image-models/timm/models/layers/activations.py
 import torch
 from torch import nn as nn
-from torch.nn import functional as F
 from torch.nn import Mish
-
+from torch.nn import functional as F
 
 __all__ = [
     "mish",

src/model_constructor/helpers.py

@@ -140,14 +140,18 @@ def print_set_fields(self) -> None:
         else:
             print("Nothing changed")
 
-    def print_changed_fields(self, show_default: bool = False, separator: str = " | ") -> None:
+    def print_changed_fields(
+        self, show_default: bool = False, separator: str = " | "
+    ) -> None:
         """Print fields changed at init."""
         if self.changed_fields:
             default_value = ""
             print("Changed fields:")
             for field in self.changed_fields:
                 if show_default:
-                    default_value = f"{separator}{self._get_str(self.model_fields[field].default)}"
+                    default_value = (
+                        f"{separator}{self._get_str(self.model_fields[field].default)}"
+                    )
                 print(f"{field}: {self._get_str_value(field)}{default_value}")
         else:
             print("Nothing changed")

src/model_constructor/model_constructor.py

@@ -1,6 +1,6 @@
 from collections import OrderedDict
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional, Union, Type
+from typing import Any, Callable, Dict, List, Optional, Type, Union
 
 from pydantic import field_validator
 from pydantic_core.core_schema import FieldValidationInfo

src/model_constructor/mxresnet.py

@@ -1,4 +1,5 @@
 from typing import List, Type
+
 from torch import nn
 
 from .xresnet import XResNet

src/model_constructor/xresnet.py

@@ -4,7 +4,7 @@
 from torch import nn
 
 from .blocks import BottleneckBlock
-from .helpers import ListStrMod, nn_seq, ModSeq
+from .helpers import ListStrMod, ModSeq, nn_seq
 from .model_constructor import ModelCfg, ModelConstructor
 
 __all__ = [

src/model_constructor/yaresnet.py

@@ -10,10 +10,9 @@
 from model_constructor.helpers import ModSeq, nn_seq
 
 from .layers import ConvBnAct, get_act
-from .model_constructor import ListStrMod, ModelConstructor, ModelCfg
+from .model_constructor import ListStrMod, ModelCfg, ModelConstructor
 from .xresnet import xresnet_stem
 
-
 __all__ = [
     "YaBasicBlock",
     "YaBottleneckBlock",