Merge pull request #122 from ayasyrev/dev

4.2

Author: ayasyrev

noxfile_cov.py

@@ -1,7 +1,7 @@
 import nox
 
 
-@nox.session(python=["3.10"])
+@nox.session(python=["3.11"])
 def cov_tests(session: nox.Session) -> None:
     args = session.posargs or ["--cov"]
     session.install(".", "pytest", "pytest-cov", "coverage[toml]")

requirements_test.txt

@@ -1,5 +1,2 @@
 pytest
-pytest-cov
-coverage[toml]
-flake8
-nox
+pytest-cov

requirements_test_extra.txt

@@ -0,0 +1,5 @@
+coverage[toml]
+black
+flake8
+nox
+isort

setup.cfg

@@ -10,15 +10,18 @@ long_description_content_type = text/markdown
 url = https://github.com/ayasyrev/model_constructor
 license = apache2
 classifiers =
-    Programming Language :: Python :: 3
+    Programming Language :: Python :: 3.8
+    Programming Language :: Python :: 3.9
+    Programming Language :: Python :: 3.10
+    Programming Language :: Python :: 3.11
     License :: OSI Approved :: Apache Software License
     Operating System :: OS Independent
 
 [options]
 package_dir =
     = src
 packages = find:
-python_requires = >=3.7
+python_requires = >=3.8, <3.12
 
 [options.packages.find]
 where = src

setup_.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/__init__.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/activations.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/helpers.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
@@ -32,7 +32,7 @@
 }
 
 
-nnModule = Union[Type[nn.Module], Callable[[], nn.Module]]
+nnModule = Union[Type[nn.Module], Callable[[Any], nn.Module]]
 
 
 class ModelCfg(Cfg, arbitrary_types_allowed=True, extra="forbid"):

src/model_constructor/model_constructor.py

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