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

nshtrainer/config/callbacks/directory_setup/__init__.py

@@ -1,13 +1,33 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __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
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.callbacks.directory_setup import (
+        CallbackConfigBase as CallbackConfigBase,
+    )
+    from nshtrainer.callbacks.directory_setup import (
+        DirectorySetupCallbackConfig as DirectorySetupCallbackConfig,
+    )
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "DirectorySetupCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.directory_setup"
+            ).DirectorySetupCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.directory_setup"
+            ).CallbackConfigBase
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/early_stopping/__init__.py

@@ -1,14 +1,38 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.callbacks.early_stopping import EarlyStoppingConfig as EarlyStoppingConfig
-from nshtrainer.callbacks.early_stopping import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.callbacks.early_stopping import MetricConfig as MetricConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.callbacks.early_stopping import (
+        CallbackConfigBase as CallbackConfigBase,
+    )
+    from nshtrainer.callbacks.early_stopping import (
+        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+    )
+    from nshtrainer.callbacks.early_stopping 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.callbacks.early_stopping"
+            ).MetricConfig
+        if name == "EarlyStoppingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.early_stopping"
+            ).EarlyStoppingCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.callbacks.early_stopping"
+            ).CallbackConfigBase
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/ema/__init__.py

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

nshtrainer/config/callbacks/finite_checks/__init__.py

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

nshtrainer/config/callbacks/gradient_skipping/__init__.py

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

nshtrainer/config/callbacks/norm_logging/__init__.py

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

nshtrainer/config/callbacks/print_table/__init__.py

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

nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py

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

nshtrainer/config/callbacks/shared_parameters/__init__.py

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

nshtrainer/config/callbacks/throughput_monitor/__init__.py

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

nshtrainer/config/callbacks/timer/__init__.py

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

nshtrainer/config/callbacks/wandb_upload_code/__init__.py

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

nshtrainer/config/callbacks/wandb_watch/__init__.py

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

nshtrainer/config/loggers/__init__.py

@@ -1,20 +1,55 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
-from nshtrainer.loggers import BaseLoggerConfig as BaseLoggerConfig
-from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
-from nshtrainer.loggers.wandb import WandbUploadCodeConfig as WandbUploadCodeConfig
-from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
-from nshtrainer.loggers.wandb import WandbWatchConfig as WandbWatchConfig
-from nshtrainer.loggers import CSVLoggerConfig as CSVLoggerConfig
-
-# Type aliases
-from nshtrainer.loggers import LoggerConfig as LoggerConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.loggers import BaseLoggerConfig as BaseLoggerConfig
+    from nshtrainer.loggers import CSVLoggerConfig as CSVLoggerConfig
+    from nshtrainer.loggers import LoggerConfig as LoggerConfig
+    from nshtrainer.loggers import TensorboardLoggerConfig as TensorboardLoggerConfig
+    from nshtrainer.loggers import WandbLoggerConfig as WandbLoggerConfig
+    from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
+    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").BaseLoggerConfig
+        if name == "TensorboardLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").TensorboardLoggerConfig
+        if name == "WandbLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").WandbLoggerConfig
+        if name == "WandbUploadCodeCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).WandbUploadCodeCallbackConfig
+        if name == "WandbWatchCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).WandbWatchCallbackConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).CallbackConfigBase
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").CSVLoggerConfig
+        if name == "LoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").LoggerConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
 
 # Submodule exports
 from . import _base as _base

nshtrainer/config/loggers/_base/__init__.py

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

nshtrainer/config/loggers/csv/__init__.py

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