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

nshtrainer/config/_directory/__init__.py

@@ -1,14 +1,26 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __codegen__ = True
 
-# Config classes
-from nshtrainer._directory import DirectoryConfig as DirectoryConfig
-from nshtrainer._directory import DirectorySetupConfig as DirectorySetupConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer._directory import DirectoryConfig as DirectoryConfig
+    from nshtrainer._directory import DirectorySetupConfig as DirectorySetupConfig
+    from nshtrainer._directory import LoggerConfig as LoggerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
 
-# Type aliases
-from nshtrainer._directory import LoggerConfig as LoggerConfig
+        if name in globals():
+            return globals()[name]
+        if name == "DirectoryConfig":
+            return importlib.import_module("nshtrainer._directory").DirectoryConfig
+        if name == "DirectorySetupConfig":
+            return importlib.import_module("nshtrainer._directory").DirectorySetupConfig
+        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,30 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer._hf_hub").CallbackConfigBase
+        if name == "HuggingFaceHubConfig":
+            return importlib.import_module("nshtrainer._hf_hub").HuggingFaceHubConfig
+        if name == "HuggingFaceHubAutoCreateConfig":
+            return importlib.import_module(
+                "nshtrainer._hf_hub"
+            ).HuggingFaceHubAutoCreateConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/__init__.py

@@ -1,35 +1,120 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+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 DirectorySetupConfig as DirectorySetupConfig
+    from nshtrainer.callbacks import EarlyStoppingConfig as EarlyStoppingConfig
+    from nshtrainer.callbacks import EMAConfig as EMAConfig
+    from nshtrainer.callbacks import EpochTimerConfig as EpochTimerConfig
+    from nshtrainer.callbacks import FiniteChecksConfig as FiniteChecksConfig
+    from nshtrainer.callbacks import GradientSkippingConfig as GradientSkippingConfig
+    from nshtrainer.callbacks import (
+        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks import NormLoggingConfig as NormLoggingConfig
+    from nshtrainer.callbacks import (
+        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+    )
+    from nshtrainer.callbacks import PrintTableMetricsConfig as PrintTableMetricsConfig
+    from nshtrainer.callbacks import RLPSanityChecksConfig as RLPSanityChecksConfig
+    from nshtrainer.callbacks import SharedParametersConfig as SharedParametersConfig
+    from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
+    from nshtrainer.callbacks import WandbUploadCodeConfig as WandbUploadCodeConfig
+    from nshtrainer.callbacks import WandbWatchConfig as WandbWatchConfig
+    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 == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer.callbacks").CallbackConfigBase
+        if name == "PrintTableMetricsConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).PrintTableMetricsConfig
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).DebugFlagCallbackConfig
+        if name == "ThroughputMonitorConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).ThroughputMonitorConfig
+        if name == "GradientSkippingConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).GradientSkippingConfig
+        if name == "RLPSanityChecksConfig":
+            return importlib.import_module("nshtrainer.callbacks").RLPSanityChecksConfig
+        if name == "WandbUploadCodeConfig":
+            return importlib.import_module("nshtrainer.callbacks").WandbUploadCodeConfig
+        if name == "MetricConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.early_stopping"
+            ).MetricConfig
+        if name == "EarlyStoppingConfig":
+            return importlib.import_module("nshtrainer.callbacks").EarlyStoppingConfig
+        if name == "WandbWatchConfig":
+            return importlib.import_module("nshtrainer.callbacks").WandbWatchConfig
+        if name == "EMAConfig":
+            return importlib.import_module("nshtrainer.callbacks").EMAConfig
+        if name == "DirectorySetupConfig":
+            return importlib.import_module("nshtrainer.callbacks").DirectorySetupConfig
+        if name == "ActSaveConfig":
+            return importlib.import_module("nshtrainer.callbacks.actsave").ActSaveConfig
+        if name == "FiniteChecksConfig":
+            return importlib.import_module("nshtrainer.callbacks").FiniteChecksConfig
+        if name == "NormLoggingConfig":
+            return importlib.import_module("nshtrainer.callbacks").NormLoggingConfig
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).LastCheckpointCallbackConfig
+        if name == "BestCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).BestCheckpointCallbackConfig
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).OnExceptionCheckpointCallbackConfig
+        if name == "EpochTimerConfig":
+            return importlib.import_module("nshtrainer.callbacks").EpochTimerConfig
+        if name == "SharedParametersConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).SharedParametersConfig
+        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}'")
 
-# Type aliases
-from nshtrainer.callbacks import CallbackConfig as CallbackConfig
 
 # Submodule exports
 from . import actsave as actsave

nshtrainer/config/callbacks/actsave/__init__.py

@@ -1,13 +1,25 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.actsave"
+            ).CallbackConfigBase
+        if name == "ActSaveConfig":
+            return importlib.import_module("nshtrainer.callbacks.actsave").ActSaveConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/base/__init__.py

@@ -1,12 +1,22 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __codegen__ = True
 
-# Config classes
-from nshtrainer.callbacks.base import CallbackConfigBase as CallbackConfigBase
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.callbacks.base import CallbackConfigBase as CallbackConfigBase
+else:
+
+    def __getattr__(name):
+        import importlib
 
-# Type aliases
+        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,68 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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 == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint"
+            ).LastCheckpointCallbackConfig
+        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
+        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,38 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CallbackConfigBase
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
+        if name == "BaseCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).BaseCheckpointCallbackConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

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

@@ -1,15 +1,45 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        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,38 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "LastCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.last_checkpoint"
+            ).LastCheckpointCallbackConfig
+        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
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

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

@@ -1,13 +1,31 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.on_exception_checkpoint"
+            ).CallbackConfigBase
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.on_exception_checkpoint"
+            ).OnExceptionCheckpointCallbackConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/debug_flag/__init__.py

@@ -1,13 +1,29 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        if name in globals():
+            return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.debug_flag"
+            ).CallbackConfigBase
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.debug_flag"
+            ).DebugFlagCallbackConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/directory_setup/__init__.py

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