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

nshtrainer/config/_directory/__init__.py

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

nshtrainer/config/_hf_hub/__init__.py

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

nshtrainer/config/callbacks/__init__.py

@@ -1,35 +1,160 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.callbacks import PrintTableMetricsConfig as PrintTableMetricsConfig
-from nshtrainer.callbacks import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
-from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
-from nshtrainer.callbacks import GradientSkippingConfig as GradientSkippingConfig
-from nshtrainer.callbacks import RLPSanityChecksConfig as RLPSanityChecksConfig
-from nshtrainer.callbacks import WandbUploadCodeConfig as WandbUploadCodeConfig
-from nshtrainer.callbacks import EarlyStoppingConfig as EarlyStoppingConfig
-from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
-from nshtrainer.callbacks import WandbWatchConfig as WandbWatchConfig
-from nshtrainer.callbacks import EMAConfig as EMAConfig
-from nshtrainer.callbacks import DirectorySetupConfig as DirectorySetupConfig
-from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
-from nshtrainer.callbacks import FiniteChecksConfig as FiniteChecksConfig
-from nshtrainer.callbacks import NormLoggingConfig as NormLoggingConfig
-from nshtrainer.callbacks import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
-from nshtrainer.callbacks import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
-from nshtrainer.callbacks import SharedParametersConfig as SharedParametersConfig
-from nshtrainer.callbacks import EpochTimerConfig as EpochTimerConfig
-from nshtrainer.callbacks import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
-from nshtrainer.callbacks.checkpoint._base import CheckpointMetadata as CheckpointMetadata
-from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
-
-# Type aliases
-from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.callbacks import (
+        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks import CallbackConfig as CallbackConfig
+    from nshtrainer.callbacks import CallbackConfigBase as CallbackConfigBase
+    from nshtrainer.callbacks import DebugFlagCallbackConfig as DebugFlagCallbackConfig
+    from nshtrainer.callbacks import (
+        DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+    )
+    from nshtrainer.callbacks import EMACallbackConfig as EMACallbackConfig
+    from nshtrainer.callbacks import (
+        EpochTimerCallbackConfig as EpochTimerCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        NormLoggingCallbackConfig as NormLoggingCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+    )
+    from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
+    from nshtrainer.callbacks import (
+        WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+    )
+    from nshtrainer.callbacks import (
+        WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+    )
+    from nshtrainer.callbacks.actsave import ActSaveConfig as ActSaveConfig
+    from nshtrainer.callbacks.checkpoint._base import (
+        BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks.checkpoint._base import (
+        CheckpointMetadata as CheckpointMetadata,
+    )
+    from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "PrintTableMetricsCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).PrintTableMetricsCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer.callbacks").CallbackConfigBase
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).DebugFlagCallbackConfig
+        if name == "ThroughputMonitorConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).ThroughputMonitorConfig
+        if name == "GradientSkippingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).GradientSkippingCallbackConfig
+        if name == "RLPSanityChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).RLPSanityChecksCallbackConfig
+        if name == "WandbUploadCodeCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).WandbUploadCodeCallbackConfig
+        if name == "MetricConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.early_stopping"
+            ).MetricConfig
+        if name == "EarlyStoppingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).EarlyStoppingCallbackConfig
+        if name == "WandbWatchCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).WandbWatchCallbackConfig
+        if name == "EMACallbackConfig":
+            return importlib.import_module("nshtrainer.callbacks").EMACallbackConfig
+        if name == "DirectorySetupCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).DirectorySetupCallbackConfig
+        if name == "ActSaveConfig":
+            return importlib.import_module("nshtrainer.callbacks.actsave").ActSaveConfig
+        if name == "FiniteChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).FiniteChecksCallbackConfig
+        if name == "NormLoggingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).NormLoggingCallbackConfig
+        if name == "EpochTimerCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).EpochTimerCallbackConfig
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).OnExceptionCheckpointCallbackConfig
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).LastCheckpointCallbackConfig
+        if name == "SharedParametersCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).SharedParametersCallbackConfig
+        if name == "BestCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).BestCheckpointCallbackConfig
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
+        if name == "BaseCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).BaseCheckpointCallbackConfig
+        if name == "CallbackConfig":
+            return importlib.import_module("nshtrainer.callbacks").CallbackConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
 
 # Submodule exports
 from . import actsave as actsave

nshtrainer/config/callbacks/actsave/__init__.py

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

nshtrainer/config/callbacks/base/__init__.py

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

nshtrainer/config/callbacks/checkpoint/__init__.py

@@ -1,19 +1,70 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.callbacks.checkpoint import LastCheckpointCallbackConfig as LastCheckpointCallbackConfig
-from nshtrainer.callbacks.checkpoint._base import CheckpointMetadata as CheckpointMetadata
-from nshtrainer.callbacks.checkpoint._base import BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig
-from nshtrainer.callbacks.checkpoint._base import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.callbacks.checkpoint import OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig
-from nshtrainer.callbacks.checkpoint import BestCheckpointCallbackConfig as BestCheckpointCallbackConfig
-from nshtrainer.callbacks.checkpoint.best_checkpoint import MetricConfig as MetricConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.callbacks.checkpoint import (
+        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks.checkpoint import (
+        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks.checkpoint import (
+        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks.checkpoint._base import (
+        BaseCheckpointCallbackConfig as BaseCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks.checkpoint._base import (
+        CallbackConfigBase as CallbackConfigBase,
+    )
+    from nshtrainer.callbacks.checkpoint._base import (
+        CheckpointMetadata as CheckpointMetadata,
+    )
+    from nshtrainer.callbacks.checkpoint.best_checkpoint import (
+        MetricConfig as MetricConfig,
+    )
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
+        if name == "BaseCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).BaseCheckpointCallbackConfig
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint"
+            ).LastCheckpointCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CallbackConfigBase
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint"
+            ).OnExceptionCheckpointCallbackConfig
+        if name == "BestCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint"
+            ).BestCheckpointCallbackConfig
+        if name == "MetricConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.best_checkpoint"
+            ).MetricConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
-# Type aliases
 
 # Submodule exports
 from . import _base as _base

nshtrainer/config/callbacks/checkpoint/_base/__init__.py

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

nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py

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

nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py

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

nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py

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

nshtrainer/config/callbacks/debug_flag/__init__.py

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