nshtrainer 1.0.0b14__py3-none-any.whl → 1.0.0b16__py3-none-any.whl

nshtrainer/configs/loggers/tensorboard/__init__.py

@@ -2,30 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.loggers.tensorboard import BaseLoggerConfig as BaseLoggerConfig
-    from nshtrainer.loggers.tensorboard import (
-        TensorboardLoggerConfig as TensorboardLoggerConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.tensorboard"
-            ).BaseLoggerConfig
-        if name == "TensorboardLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.tensorboard"
-            ).TensorboardLoggerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.loggers.tensorboard import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.tensorboard import (
+    TensorboardLoggerConfig as TensorboardLoggerConfig,
+)

nshtrainer/configs/loggers/wandb/__init__.py

@@ -2,43 +2,12 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.loggers.wandb import BaseLoggerConfig as BaseLoggerConfig
-    from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
-    from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
-    from nshtrainer.loggers.wandb import (
-        WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
-    )
-    from nshtrainer.loggers.wandb import (
-        WandbWatchCallbackConfig as WandbWatchCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.wandb").BaseLoggerConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).CallbackConfigBase
-        if name == "WandbLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.wandb").WandbLoggerConfig
-        if name == "WandbUploadCodeCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).WandbUploadCodeCallbackConfig
-        if name == "WandbWatchCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).WandbWatchCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.loggers.wandb import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
+from nshtrainer.loggers.wandb import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)
+from nshtrainer.loggers.wandb import (
+    WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+)

nshtrainer/configs/lr_scheduler/__init__.py

@@ -2,58 +2,17 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
+from nshtrainer.lr_scheduler import (
+    LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
+)
+from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
+from nshtrainer.lr_scheduler import LRSchedulerConfigBase as LRSchedulerConfigBase
+from nshtrainer.lr_scheduler import ReduceLROnPlateauConfig as ReduceLROnPlateauConfig
+from nshtrainer.lr_scheduler.linear_warmup_cosine import (
+    DurationConfig as DurationConfig,
+)
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import MetricConfig as MetricConfig
 
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.lr_scheduler import (
-        LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
-    )
-    from nshtrainer.lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
-    from nshtrainer.lr_scheduler import LRSchedulerConfigBase as LRSchedulerConfigBase
-    from nshtrainer.lr_scheduler import (
-        ReduceLROnPlateauConfig as ReduceLROnPlateauConfig,
-    )
-    from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-        DurationConfig as DurationConfig,
-    )
-    from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
-        MetricConfig as MetricConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "LRSchedulerConfigBase":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler"
-            ).LRSchedulerConfigBase
-        if name == "LinearWarmupCosineDecayLRSchedulerConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler"
-            ).LinearWarmupCosineDecayLRSchedulerConfig
-        if name == "MetricConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
-            ).MetricConfig
-        if name == "ReduceLROnPlateauConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler"
-            ).ReduceLROnPlateauConfig
-        if name == "DurationConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.linear_warmup_cosine"
-            ).DurationConfig
-        if name == "LRSchedulerConfig":
-            return importlib.import_module("nshtrainer.lr_scheduler").LRSchedulerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-
-# Submodule exports
 from . import _base as _base
 from . import linear_warmup_cosine as linear_warmup_cosine
 from . import reduce_lr_on_plateau as reduce_lr_on_plateau

nshtrainer/configs/lr_scheduler/_base/__init__.py

@@ -2,25 +2,4 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.lr_scheduler._base import (
-        LRSchedulerConfigBase as LRSchedulerConfigBase,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "LRSchedulerConfigBase":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler._base"
-            ).LRSchedulerConfigBase
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.lr_scheduler._base import LRSchedulerConfigBase as LRSchedulerConfigBase

nshtrainer/configs/lr_scheduler/linear_warmup_cosine/__init__.py

@@ -2,39 +2,12 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-        DurationConfig as DurationConfig,
-    )
-    from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-        LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
-    )
-    from nshtrainer.lr_scheduler.linear_warmup_cosine import (
-        LRSchedulerConfigBase as LRSchedulerConfigBase,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "LRSchedulerConfigBase":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.linear_warmup_cosine"
-            ).LRSchedulerConfigBase
-        if name == "LinearWarmupCosineDecayLRSchedulerConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.linear_warmup_cosine"
-            ).LinearWarmupCosineDecayLRSchedulerConfig
-        if name == "DurationConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.linear_warmup_cosine"
-            ).DurationConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.lr_scheduler.linear_warmup_cosine import (
+    DurationConfig as DurationConfig,
+)
+from nshtrainer.lr_scheduler.linear_warmup_cosine import (
+    LinearWarmupCosineDecayLRSchedulerConfig as LinearWarmupCosineDecayLRSchedulerConfig,
+)
+from nshtrainer.lr_scheduler.linear_warmup_cosine import (
+    LRSchedulerConfigBase as LRSchedulerConfigBase,
+)

nshtrainer/configs/lr_scheduler/reduce_lr_on_plateau/__init__.py

@@ -2,39 +2,10 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
-        LRSchedulerConfigBase as LRSchedulerConfigBase,
-    )
-    from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
-        MetricConfig as MetricConfig,
-    )
-    from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
-        ReduceLROnPlateauConfig as ReduceLROnPlateauConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "LRSchedulerConfigBase":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
-            ).LRSchedulerConfigBase
-        if name == "MetricConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
-            ).MetricConfig
-        if name == "ReduceLROnPlateauConfig":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
-            ).ReduceLROnPlateauConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
+    LRSchedulerConfigBase as LRSchedulerConfigBase,
+)
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import MetricConfig as MetricConfig
+from nshtrainer.lr_scheduler.reduce_lr_on_plateau import (
+    ReduceLROnPlateauConfig as ReduceLROnPlateauConfig,
+)

nshtrainer/configs/metrics/__init__.py

@@ -2,23 +2,6 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
+from nshtrainer.metrics import MetricConfig as MetricConfig
 
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.metrics import MetricConfig as MetricConfig
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.metrics").MetricConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-
-# Submodule exports
 from . import _config as _config

nshtrainer/configs/metrics/_config/__init__.py

@@ -2,21 +2,4 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.metrics._config import MetricConfig as MetricConfig
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.metrics._config").MetricConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.metrics._config import MetricConfig as MetricConfig

nshtrainer/configs/nn/__init__.py

@@ -2,76 +2,25 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
+from nshtrainer.nn import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn import ELUNonlinearityConfig as ELUNonlinearityConfig
+from nshtrainer.nn import GELUNonlinearityConfig as GELUNonlinearityConfig
+from nshtrainer.nn import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
+from nshtrainer.nn import MishNonlinearityConfig as MishNonlinearityConfig
+from nshtrainer.nn import MLPConfig as MLPConfig
+from nshtrainer.nn import NonlinearityConfig as NonlinearityConfig
+from nshtrainer.nn import PReLUConfig as PReLUConfig
+from nshtrainer.nn import ReLUNonlinearityConfig as ReLUNonlinearityConfig
+from nshtrainer.nn import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
+from nshtrainer.nn import SiLUNonlinearityConfig as SiLUNonlinearityConfig
+from nshtrainer.nn import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
+from nshtrainer.nn import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
+from nshtrainer.nn import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
+from nshtrainer.nn import SwishNonlinearityConfig as SwishNonlinearityConfig
+from nshtrainer.nn import TanhNonlinearityConfig as TanhNonlinearityConfig
+from nshtrainer.nn.nonlinearity import (
+    SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
+)
 
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.nn import BaseNonlinearityConfig as BaseNonlinearityConfig
-    from nshtrainer.nn import ELUNonlinearityConfig as ELUNonlinearityConfig
-    from nshtrainer.nn import GELUNonlinearityConfig as GELUNonlinearityConfig
-    from nshtrainer.nn import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
-    from nshtrainer.nn import MishNonlinearityConfig as MishNonlinearityConfig
-    from nshtrainer.nn import MLPConfig as MLPConfig
-    from nshtrainer.nn import NonlinearityConfig as NonlinearityConfig
-    from nshtrainer.nn import PReLUConfig as PReLUConfig
-    from nshtrainer.nn import ReLUNonlinearityConfig as ReLUNonlinearityConfig
-    from nshtrainer.nn import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
-    from nshtrainer.nn import SiLUNonlinearityConfig as SiLUNonlinearityConfig
-    from nshtrainer.nn import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
-    from nshtrainer.nn import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
-    from nshtrainer.nn import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
-    from nshtrainer.nn import SwishNonlinearityConfig as SwishNonlinearityConfig
-    from nshtrainer.nn import TanhNonlinearityConfig as TanhNonlinearityConfig
-    from nshtrainer.nn.nonlinearity import (
-        SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").BaseNonlinearityConfig
-        if name == "ELUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").ELUNonlinearityConfig
-        if name == "GELUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").GELUNonlinearityConfig
-        if name == "LeakyReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").LeakyReLUNonlinearityConfig
-        if name == "MLPConfig":
-            return importlib.import_module("nshtrainer.nn").MLPConfig
-        if name == "MishNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").MishNonlinearityConfig
-        if name == "PReLUConfig":
-            return importlib.import_module("nshtrainer.nn").PReLUConfig
-        if name == "ReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").ReLUNonlinearityConfig
-        if name == "SiLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SiLUNonlinearityConfig
-        if name == "SigmoidNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SigmoidNonlinearityConfig
-        if name == "SoftmaxNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftmaxNonlinearityConfig
-        if name == "SoftplusNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftplusNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftsignNonlinearityConfig
-        if name == "SwiGLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SwiGLUNonlinearityConfig
-        if name == "SwishNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SwishNonlinearityConfig
-        if name == "TanhNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").TanhNonlinearityConfig
-        if name == "NonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").NonlinearityConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-
-# Submodule exports
 from . import mlp as mlp
 from . import nonlinearity as nonlinearity

nshtrainer/configs/nn/mlp/__init__.py

@@ -2,27 +2,6 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.nn.mlp import BaseNonlinearityConfig as BaseNonlinearityConfig
-    from nshtrainer.nn.mlp import MLPConfig as MLPConfig
-    from nshtrainer.nn.mlp import NonlinearityConfig as NonlinearityConfig
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn.mlp").BaseNonlinearityConfig
-        if name == "MLPConfig":
-            return importlib.import_module("nshtrainer.nn.mlp").MLPConfig
-        if name == "NonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn.mlp").NonlinearityConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.nn.mlp import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn.mlp import MLPConfig as MLPConfig
+from nshtrainer.nn.mlp import NonlinearityConfig as NonlinearityConfig

nshtrainer/configs/nn/nonlinearity/__init__.py

@@ -2,124 +2,33 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.nn.nonlinearity import (
-        BaseNonlinearityConfig as BaseNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        ELUNonlinearityConfig as ELUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        GELUNonlinearityConfig as GELUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        MishNonlinearityConfig as MishNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
-    from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
-    from nshtrainer.nn.nonlinearity import (
-        ReLUNonlinearityConfig as ReLUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SigmoidNonlinearityConfig as SigmoidNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SiLUNonlinearityConfig as SiLUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SoftplusNonlinearityConfig as SoftplusNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SoftsignNonlinearityConfig as SoftsignNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        SwishNonlinearityConfig as SwishNonlinearityConfig,
-    )
-    from nshtrainer.nn.nonlinearity import (
-        TanhNonlinearityConfig as TanhNonlinearityConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).BaseNonlinearityConfig
-        if name == "ELUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).ELUNonlinearityConfig
-        if name == "GELUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).GELUNonlinearityConfig
-        if name == "LeakyReLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).LeakyReLUNonlinearityConfig
-        if name == "MishNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).MishNonlinearityConfig
-        if name == "PReLUConfig":
-            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
-        if name == "ReLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).ReLUNonlinearityConfig
-        if name == "SiLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SiLUNonlinearityConfig
-        if name == "SigmoidNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SigmoidNonlinearityConfig
-        if name == "SoftmaxNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SoftmaxNonlinearityConfig
-        if name == "SoftplusNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SoftplusNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SoftsignNonlinearityConfig
-        if name == "SwiGLUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SwiGLUNonlinearityConfig
-        if name == "SwishNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SwishNonlinearityConfig
-        if name == "TanhNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).TanhNonlinearityConfig
-        if name == "NonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).NonlinearityConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.nn.nonlinearity import BaseNonlinearityConfig as BaseNonlinearityConfig
+from nshtrainer.nn.nonlinearity import ELUNonlinearityConfig as ELUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import GELUNonlinearityConfig as GELUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import (
+    LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearityConfig
+from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
+from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
+from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import (
+    SigmoidNonlinearityConfig as SigmoidNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
+from nshtrainer.nn.nonlinearity import (
+    SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import (
+    SoftplusNonlinearityConfig as SoftplusNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import (
+    SoftsignNonlinearityConfig as SoftsignNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import (
+    SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import (
+    SwishNonlinearityConfig as SwishNonlinearityConfig,
+)
+from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig

nshtrainer/configs/optimizer/__init__.py

@@ -2,27 +2,6 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.optimizer import AdamWConfig as AdamWConfig
-    from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
-    from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "AdamWConfig":
-            return importlib.import_module("nshtrainer.optimizer").AdamWConfig
-        if name == "OptimizerConfigBase":
-            return importlib.import_module("nshtrainer.optimizer").OptimizerConfigBase
-        if name == "OptimizerConfig":
-            return importlib.import_module("nshtrainer.optimizer").OptimizerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.optimizer import AdamWConfig as AdamWConfig
+from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
+from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase