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

nshtrainer/config/callbacks/early_stopping/__init__.py

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

nshtrainer/config/callbacks/ema/__init__.py

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

nshtrainer/config/callbacks/finite_checks/__init__.py

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

nshtrainer/config/callbacks/gradient_skipping/__init__.py

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

nshtrainer/config/callbacks/norm_logging/__init__.py

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

nshtrainer/config/callbacks/print_table/__init__.py

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

nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py

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

nshtrainer/config/callbacks/shared_parameters/__init__.py

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

nshtrainer/config/callbacks/throughput_monitor/__init__.py

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

nshtrainer/config/callbacks/timer/__init__.py

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

nshtrainer/config/callbacks/wandb_upload_code/__init__.py

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

nshtrainer/config/callbacks/wandb_watch/__init__.py

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

nshtrainer/config/loggers/__init__.py

@@ -1,20 +1,47 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+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 WandbUploadCodeConfig as WandbUploadCodeConfig
+    from nshtrainer.loggers.wandb import WandbWatchConfig as WandbWatchConfig
+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 == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).CallbackConfigBase
+        if name == "WandbLoggerConfig":
+            return importlib.import_module("nshtrainer.loggers").WandbLoggerConfig
+        if name == "WandbUploadCodeConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).WandbUploadCodeConfig
+        if name == "WandbWatchConfig":
+            return importlib.import_module("nshtrainer.loggers.wandb").WandbWatchConfig
+        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}'")
 
-# Type aliases
-from nshtrainer.loggers import LoggerConfig as LoggerConfig
 
 # Submodule exports
 from . import _base as _base

nshtrainer/config/loggers/_base/__init__.py

@@ -1,12 +1,20 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __codegen__ = True
 
-# Config classes
-from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig
+else:
+
+    def __getattr__(name):
+        import importlib
 
-# Type aliases
+        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,23 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
-
 __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
+
+# 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
 
-# Type aliases
+        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

nshtrainer/config/loggers/tensorboard/__init__.py

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