nshtrainer 0.41.1__py3-none-any.whl → 0.43.0__py3-none-any.whl

nshtrainer/config/nn/nonlinearity/__init__.py

@@ -1,27 +1,125 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.nn.nonlinearity import TanhNonlinearityConfig as TanhNonlinearityConfig
-from nshtrainer.nn.nonlinearity import ReLUNonlinearityConfig as ReLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import PReLUConfig as PReLUConfig
-from nshtrainer.nn.nonlinearity import SiLUNonlinearityConfig as SiLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import LeakyReLUNonlinearityConfig as LeakyReLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SoftsignNonlinearityConfig as SoftsignNonlinearityConfig
-from nshtrainer.nn.nonlinearity import MishNonlinearityConfig as MishNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SigmoidNonlinearityConfig as SigmoidNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SoftplusNonlinearityConfig as SoftplusNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SwiGLUNonlinearityConfig as SwiGLUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import ELUNonlinearityConfig as ELUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import BaseNonlinearityConfig as BaseNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SoftmaxNonlinearityConfig as SoftmaxNonlinearityConfig
-from nshtrainer.nn.nonlinearity import GELUNonlinearityConfig as GELUNonlinearityConfig
-from nshtrainer.nn.nonlinearity import SwishNonlinearityConfig as SwishNonlinearityConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
-from nshtrainer.nn.nonlinearity import NonlinearityConfig as NonlinearityConfig
+# 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 == "PReLUConfig":
+            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
+        if name == "LeakyReLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).LeakyReLUNonlinearityConfig
+        if name == "SwiGLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SwiGLUNonlinearityConfig
+        if name == "SoftsignNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SoftsignNonlinearityConfig
+        if name == "SiLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SiLUNonlinearityConfig
+        if name == "SigmoidNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SigmoidNonlinearityConfig
+        if name == "SoftplusNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SoftplusNonlinearityConfig
+        if name == "ELUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).ELUNonlinearityConfig
+        if name == "SoftmaxNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SoftmaxNonlinearityConfig
+        if name == "GELUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).GELUNonlinearityConfig
+        if name == "SwishNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).SwishNonlinearityConfig
+        if name == "MishNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).MishNonlinearityConfig
+        if name == "TanhNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).TanhNonlinearityConfig
+        if name == "ReLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).ReLUNonlinearityConfig
+        if name == "BaseNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).BaseNonlinearityConfig
+        if name == "NonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).NonlinearityConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/optimizer/__init__.py

@@ -1,14 +1,28 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.optimizer import OptimizerConfigBase as OptimizerConfigBase
-from nshtrainer.optimizer import AdamWConfig as AdamWConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
-from nshtrainer.optimizer import OptimizerConfig as OptimizerConfig
+# 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 == "OptimizerConfigBase":
+            return importlib.import_module("nshtrainer.optimizer").OptimizerConfigBase
+        if name == "AdamWConfig":
+            return importlib.import_module("nshtrainer.optimizer").AdamWConfig
+        if name == "OptimizerConfig":
+            return importlib.import_module("nshtrainer.optimizer").OptimizerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/__init__.py

@@ -1,17 +1,36 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
-from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
-from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
-from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
+    from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
+    from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
+    from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
+    from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "BaseProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").BaseProfilerConfig
+        if name == "PyTorchProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").PyTorchProfilerConfig
+        if name == "AdvancedProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").AdvancedProfilerConfig
+        if name == "SimpleProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").SimpleProfilerConfig
+        if name == "ProfilerConfig":
+            return importlib.import_module("nshtrainer.profiler").ProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
-# Type aliases
-from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
 
 # Submodule exports
 from . import _base as _base

nshtrainer/config/profiler/_base/__init__.py

@@ -1,12 +1,24 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.profiler._base import BaseProfilerConfig as BaseProfilerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler._base"
+            ).BaseProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/advanced/__init__.py

@@ -1,13 +1,31 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.profiler.advanced import AdvancedProfilerConfig as AdvancedProfilerConfig
-from nshtrainer.profiler.advanced import BaseProfilerConfig as BaseProfilerConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.profiler.advanced import (
+        AdvancedProfilerConfig as AdvancedProfilerConfig,
+    )
+    from nshtrainer.profiler.advanced import BaseProfilerConfig as BaseProfilerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "AdvancedProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.advanced"
+            ).AdvancedProfilerConfig
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.advanced"
+            ).BaseProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/pytorch/__init__.py

@@ -1,13 +1,31 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.profiler.pytorch import PyTorchProfilerConfig as PyTorchProfilerConfig
-from nshtrainer.profiler.pytorch import BaseProfilerConfig as BaseProfilerConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.profiler.pytorch import BaseProfilerConfig as BaseProfilerConfig
+    from nshtrainer.profiler.pytorch import (
+        PyTorchProfilerConfig as PyTorchProfilerConfig,
+    )
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "PyTorchProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.pytorch"
+            ).PyTorchProfilerConfig
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.pytorch"
+            ).BaseProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/simple/__init__.py

@@ -1,13 +1,29 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.profiler.simple import BaseProfilerConfig as BaseProfilerConfig
-from nshtrainer.profiler.simple import SimpleProfilerConfig as SimpleProfilerConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.profiler.simple import BaseProfilerConfig as BaseProfilerConfig
+    from nshtrainer.profiler.simple import SimpleProfilerConfig as SimpleProfilerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "SimpleProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.simple"
+            ).SimpleProfilerConfig
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.simple"
+            ).BaseProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/runner/__init__.py

@@ -1,12 +1,22 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.runner import BaseConfig as BaseConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.runner import BaseConfig as BaseConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "BaseConfig":
+            return importlib.import_module("nshtrainer.runner").BaseConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/trainer/_config/__init__.py

@@ -1,35 +1,153 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.trainer._config import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
-from nshtrainer.trainer._config import TensorboardLoggerConfig as TensorboardLoggerConfig
-from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
-from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
-from nshtrainer.trainer._config import SharedParametersConfig as SharedParametersConfig
-from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
-from nshtrainer.trainer._config import ReproducibilityConfig as ReproducibilityConfig
-from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
-from nshtrainer.trainer._config import CheckpointLoadingConfig as CheckpointLoadingConfig
-from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
-from nshtrainer.trainer._config import LoggingConfig as LoggingConfig
-from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.trainer._config import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
-from nshtrainer.trainer._config import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
-from nshtrainer.trainer._config import RLPSanityChecksConfig as RLPSanityChecksConfig
-from nshtrainer.trainer._config import EarlyStoppingConfig as EarlyStoppingConfig
-from nshtrainer.trainer._config import OptimizationConfig as OptimizationConfig
-from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
-from nshtrainer.trainer._config import DebugFlagCallbackConfig as DebugFlagCallbackConfig
-from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
-from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
-from nshtrainer.trainer._config import CheckpointCallbackConfig as CheckpointCallbackConfig
-from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
-from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.trainer._config import (
+        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
+    from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
+    from nshtrainer.trainer._config import (
+        CheckpointCallbackConfig as CheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        CheckpointLoadingConfig as CheckpointLoadingConfig,
+    )
+    from nshtrainer.trainer._config import (
+        CheckpointSavingConfig as CheckpointSavingConfig,
+    )
+    from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
+    from nshtrainer.trainer._config import (
+        DebugFlagCallbackConfig as DebugFlagCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        GradientClippingConfig as GradientClippingConfig,
+    )
+    from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
+    from nshtrainer.trainer._config import (
+        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
+    from nshtrainer.trainer._config import LoggingConfig as LoggingConfig
+    from nshtrainer.trainer._config import (
+        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+    )
+    from nshtrainer.trainer._config import OptimizationConfig as OptimizationConfig
+    from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+    from nshtrainer.trainer._config import (
+        ReproducibilityConfig as ReproducibilityConfig,
+    )
+    from nshtrainer.trainer._config import (
+        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+    )
+    from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+    from nshtrainer.trainer._config import (
+        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+    )
+    from nshtrainer.trainer._config import (
+        TensorboardLoggerConfig as TensorboardLoggerConfig,
+    )
+    from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
+    from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "SanityCheckingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).SanityCheckingConfig
+        if name == "TrainerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").TrainerConfig
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).OnExceptionCheckpointCallbackConfig
+        if name == "GradientClippingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).GradientClippingConfig
+        if name == "WandbLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).WandbLoggerConfig
+        if name == "LoggingConfig":
+            return importlib.import_module("nshtrainer.trainer._config").LoggingConfig
+        if name == "TensorboardLoggerConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).TensorboardLoggerConfig
+        if name == "RLPSanityChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).RLPSanityChecksCallbackConfig
+        if name == "CheckpointSavingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointSavingConfig
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
+        if name == "HuggingFaceHubConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).HuggingFaceHubConfig
+        if name == "CheckpointLoadingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointLoadingConfig
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).DebugFlagCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CallbackConfigBase
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).LastCheckpointCallbackConfig
+        if name == "SharedParametersCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).SharedParametersCallbackConfig
+        if name == "ReproducibilityConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).ReproducibilityConfig
+        if name == "EarlyStoppingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).EarlyStoppingCallbackConfig
+        if name == "OptimizationConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).OptimizationConfig
+        if name == "BestCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).BestCheckpointCallbackConfig
+        if name == "CallbackConfig":
+            return importlib.import_module("nshtrainer.trainer._config").CallbackConfig
+        if name == "CheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).CheckpointCallbackConfig
+        if name == "LoggerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").LoggerConfig
+        if name == "ProfilerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").ProfilerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/trainer/checkpoint_connector/__init__.py

@@ -1,12 +1,26 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.trainer.checkpoint_connector import CheckpointLoadingConfig as CheckpointLoadingConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.trainer.checkpoint_connector import (
+        CheckpointLoadingConfig as CheckpointLoadingConfig,
+    )
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "CheckpointLoadingConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer.checkpoint_connector"
+            ).CheckpointLoadingConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports