nshtrainer 1.0.0b14__py3-none-any.whl → 1.0.0b16__py3-none-any.whl

nshtrainer/configs/callbacks/early_stopping/__init__.py

@@ -2,37 +2,8 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        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 == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.early_stopping"
-            ).CallbackConfigBase
-        if name == "EarlyStoppingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.early_stopping"
-            ).EarlyStoppingCallbackConfig
-        if name == "MetricConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.early_stopping"
-            ).MetricConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+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

nshtrainer/configs/callbacks/ema/__init__.py

@@ -2,26 +2,5 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# 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 == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.ema"
-            ).CallbackConfigBase
-        if name == "EMACallbackConfig":
-            return importlib.import_module("nshtrainer.callbacks.ema").EMACallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.ema import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.ema import EMACallbackConfig as EMACallbackConfig

nshtrainer/configs/callbacks/finite_checks/__init__.py

@@ -2,32 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.finite_checks"
-            ).CallbackConfigBase
-        if name == "FiniteChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.finite_checks"
-            ).FiniteChecksCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.finite_checks import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.finite_checks import (
+    FiniteChecksCallbackConfig as FiniteChecksCallbackConfig,
+)

nshtrainer/configs/callbacks/gradient_skipping/__init__.py

@@ -2,32 +2,9 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.gradient_skipping"
-            ).CallbackConfigBase
-        if name == "GradientSkippingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.gradient_skipping"
-            ).GradientSkippingCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.gradient_skipping import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.gradient_skipping import (
+    GradientSkippingCallbackConfig as GradientSkippingCallbackConfig,
+)

nshtrainer/configs/callbacks/log_epoch/__init__.py

@@ -2,30 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.callbacks.log_epoch import CallbackConfigBase as CallbackConfigBase
-    from nshtrainer.callbacks.log_epoch import (
-        LogEpochCallbackConfig as LogEpochCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.log_epoch"
-            ).CallbackConfigBase
-        if name == "LogEpochCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.log_epoch"
-            ).LogEpochCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.log_epoch import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.log_epoch import (
+    LogEpochCallbackConfig as LogEpochCallbackConfig,
+)

nshtrainer/configs/callbacks/lr_monitor/__init__.py

@@ -2,30 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.callbacks.lr_monitor import CallbackConfigBase as CallbackConfigBase
-    from nshtrainer.callbacks.lr_monitor import (
-        LearningRateMonitorConfig as LearningRateMonitorConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.lr_monitor"
-            ).CallbackConfigBase
-        if name == "LearningRateMonitorConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.lr_monitor"
-            ).LearningRateMonitorConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.lr_monitor import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.lr_monitor import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)

nshtrainer/configs/callbacks/norm_logging/__init__.py

@@ -2,32 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        NormLoggingCallbackConfig as NormLoggingCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.norm_logging"
-            ).CallbackConfigBase
-        if name == "NormLoggingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.norm_logging"
-            ).NormLoggingCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.norm_logging import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.norm_logging import (
+    NormLoggingCallbackConfig as NormLoggingCallbackConfig,
+)

nshtrainer/configs/callbacks/print_table/__init__.py

@@ -2,32 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.print_table"
-            ).CallbackConfigBase
-        if name == "PrintTableMetricsCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.print_table"
-            ).PrintTableMetricsCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.print_table import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.print_table import (
+    PrintTableMetricsCallbackConfig as PrintTableMetricsCallbackConfig,
+)

nshtrainer/configs/callbacks/rlp_sanity_checks/__init__.py

@@ -2,32 +2,9 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.rlp_sanity_checks"
-            ).CallbackConfigBase
-        if name == "RLPSanityChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.rlp_sanity_checks"
-            ).RLPSanityChecksCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.rlp_sanity_checks import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.rlp_sanity_checks import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)

nshtrainer/configs/callbacks/shared_parameters/__init__.py

@@ -2,32 +2,9 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.shared_parameters"
-            ).CallbackConfigBase
-        if name == "SharedParametersCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.shared_parameters"
-            ).SharedParametersCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.shared_parameters import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.shared_parameters import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)

nshtrainer/configs/callbacks/timer/__init__.py

@@ -2,30 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# 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 == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.timer"
-            ).CallbackConfigBase
-        if name == "EpochTimerCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.timer"
-            ).EpochTimerCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.timer import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.timer import (
+    EpochTimerCallbackConfig as EpochTimerCallbackConfig,
+)

nshtrainer/configs/callbacks/wandb_upload_code/__init__.py

@@ -2,32 +2,9 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_upload_code"
-            ).CallbackConfigBase
-        if name == "WandbUploadCodeCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_upload_code"
-            ).WandbUploadCodeCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.wandb_upload_code import (
+    CallbackConfigBase as CallbackConfigBase,
+)
+from nshtrainer.callbacks.wandb_upload_code import (
+    WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+)

nshtrainer/configs/callbacks/wandb_watch/__init__.py

@@ -2,32 +2,7 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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 (
-        WandbWatchCallbackConfig as WandbWatchCallbackConfig,
-    )
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_watch"
-            ).CallbackConfigBase
-        if name == "WandbWatchCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_watch"
-            ).WandbWatchCallbackConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.callbacks.wandb_watch import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.callbacks.wandb_watch import (
+    WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+)

nshtrainer/configs/loggers/__init__.py

@@ -2,59 +2,20 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
+from nshtrainer.loggers import ActSaveLoggerConfig as ActSaveLoggerConfig
+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,
+)
 
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.loggers import ActSaveLoggerConfig as ActSaveLoggerConfig
-    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 == "ActSaveLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers").ActSaveLoggerConfig
-        if name == "BaseLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers").BaseLoggerConfig
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers").CSVLoggerConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).CallbackConfigBase
-        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 == "LoggerConfig":
-            return importlib.import_module("nshtrainer.loggers").LoggerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-
-# Submodule exports
 from . import _base as _base
 from . import actsave as actsave
 from . import csv as csv

nshtrainer/configs/loggers/_base/__init__.py

@@ -2,21 +2,4 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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
-
-        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
+from nshtrainer.loggers._base import BaseLoggerConfig as BaseLoggerConfig

nshtrainer/configs/loggers/actsave/__init__.py

@@ -2,28 +2,5 @@ from __future__ import annotations
 
 __codegen__ = True
 
-from typing import TYPE_CHECKING
-
-# Config/alias imports
-
-if TYPE_CHECKING:
-    from nshtrainer.loggers.actsave import ActSaveLoggerConfig as ActSaveLoggerConfig
-    from nshtrainer.loggers.actsave import BaseLoggerConfig as BaseLoggerConfig
-else:
-
-    def __getattr__(name):
-        import importlib
-
-        if name in globals():
-            return globals()[name]
-        if name == "ActSaveLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.actsave"
-            ).ActSaveLoggerConfig
-        if name == "BaseLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.loggers.actsave"
-            ).BaseLoggerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.loggers.actsave import ActSaveLoggerConfig as ActSaveLoggerConfig
+from nshtrainer.loggers.actsave import BaseLoggerConfig as BaseLoggerConfig

nshtrainer/configs/loggers/csv/__init__.py

@@ -2,24 +2,5 @@ from __future__ import annotations
 
 __codegen__ = True
 
-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
-
-        if name in globals():
-            return globals()[name]
-        if name == "BaseLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.csv").BaseLoggerConfig
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.loggers.csv").CSVLoggerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-
-# Submodule exports
+from nshtrainer.loggers.csv import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.loggers.csv import CSVLoggerConfig as CSVLoggerConfig