PyPI - nshtrainer - Versions diffs - 1.0.0b14__py3-none-any.whl → 1.0.0b16__py3-none-any.whl - Mend

nshtrainer 1.0.0b14py3-none-any.whl → 1.0.0b16py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (61) hide show

nshtrainer/configs/__init__.py +134 -405
nshtrainer/configs/_checkpoint/__init__.py +2 -25
nshtrainer/configs/_checkpoint/metadata/__init__.py +2 -25
nshtrainer/configs/_directory/__init__.py +5 -28
nshtrainer/configs/_hf_hub/__init__.py +5 -28
nshtrainer/configs/callbacks/__init__.py +52 -161
nshtrainer/configs/callbacks/actsave/__init__.py +2 -23
nshtrainer/configs/callbacks/base/__init__.py +1 -20
nshtrainer/configs/callbacks/checkpoint/__init__.py +19 -64
nshtrainer/configs/callbacks/checkpoint/_base/__init__.py +9 -36
nshtrainer/configs/callbacks/checkpoint/best_checkpoint/__init__.py +10 -43
nshtrainer/configs/callbacks/checkpoint/last_checkpoint/__init__.py +9 -36
nshtrainer/configs/callbacks/checkpoint/on_exception_checkpoint/__init__.py +6 -29
nshtrainer/configs/callbacks/debug_flag/__init__.py +4 -27
nshtrainer/configs/callbacks/directory_setup/__init__.py +6 -29
nshtrainer/configs/callbacks/early_stopping/__init__.py +5 -34
nshtrainer/configs/callbacks/ema/__init__.py +2 -23
nshtrainer/configs/callbacks/finite_checks/__init__.py +4 -29
nshtrainer/configs/callbacks/gradient_skipping/__init__.py +6 -29
nshtrainer/configs/callbacks/log_epoch/__init__.py +4 -27
nshtrainer/configs/callbacks/lr_monitor/__init__.py +4 -27
nshtrainer/configs/callbacks/norm_logging/__init__.py +4 -29
nshtrainer/configs/callbacks/print_table/__init__.py +4 -29
nshtrainer/configs/callbacks/rlp_sanity_checks/__init__.py +6 -29
nshtrainer/configs/callbacks/shared_parameters/__init__.py +6 -29
nshtrainer/configs/callbacks/timer/__init__.py +4 -27
nshtrainer/configs/callbacks/wandb_upload_code/__init__.py +6 -29
nshtrainer/configs/callbacks/wandb_watch/__init__.py +4 -29
nshtrainer/configs/loggers/__init__.py +13 -52
nshtrainer/configs/loggers/_base/__init__.py +1 -18
nshtrainer/configs/loggers/actsave/__init__.py +2 -25
nshtrainer/configs/loggers/csv/__init__.py +2 -21
nshtrainer/configs/loggers/tensorboard/__init__.py +4 -27
nshtrainer/configs/loggers/wandb/__init__.py +9 -40
nshtrainer/configs/lr_scheduler/__init__.py +10 -51
nshtrainer/configs/lr_scheduler/_base/__init__.py +1 -22
nshtrainer/configs/lr_scheduler/linear_warmup_cosine/__init__.py +9 -36
nshtrainer/configs/lr_scheduler/reduce_lr_on_plateau/__init__.py +7 -36
nshtrainer/configs/metrics/__init__.py +1 -18
nshtrainer/configs/metrics/_config/__init__.py +1 -18
nshtrainer/configs/nn/__init__.py +19 -70
nshtrainer/configs/nn/mlp/__init__.py +3 -24
nshtrainer/configs/nn/nonlinearity/__init__.py +30 -121
nshtrainer/configs/optimizer/__init__.py +3 -24
nshtrainer/configs/profiler/__init__.py +5 -30
nshtrainer/configs/profiler/_base/__init__.py +1 -20
nshtrainer/configs/profiler/advanced/__init__.py +4 -27
nshtrainer/configs/profiler/pytorch/__init__.py +2 -27
nshtrainer/configs/profiler/simple/__init__.py +2 -25
nshtrainer/configs/trainer/__init__.py +50 -169
nshtrainer/configs/trainer/_config/__init__.py +50 -169
nshtrainer/configs/trainer/trainer/__init__.py +2 -23
nshtrainer/configs/util/__init__.py +33 -102
nshtrainer/configs/util/_environment_info/__init__.py +29 -90
nshtrainer/configs/util/config/__init__.py +4 -27
nshtrainer/configs/util/config/dtype/__init__.py +1 -18
nshtrainer/configs/util/config/duration/__init__.py +3 -30
nshtrainer/trainer/_config.py +42 -10
{nshtrainer-1.0.0b14.dist-info → nshtrainer-1.0.0b16.dist-info}/METADATA +1 -1
{nshtrainer-1.0.0b14.dist-info → nshtrainer-1.0.0b16.dist-info}/RECORD +61 -61
{nshtrainer-1.0.0b14.dist-info → nshtrainer-1.0.0b16.dist-info}/WHEEL +0 -0

nshtrainer/configs/profiler/__init__.py CHANGED Viewed

@@ -2,37 +2,12 @@ from __future__ import annotations
 __codegen__ = True
-from typing import TYPE_CHECKING
+from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
+from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
+from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
+from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
+from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
-# Config/alias imports
-if TYPE_CHECKING:
-    from nshtrainer.profiler import AdvancedProfilerConfig as AdvancedProfilerConfig
-    from nshtrainer.profiler import BaseProfilerConfig as BaseProfilerConfig
-    from nshtrainer.profiler import ProfilerConfig as ProfilerConfig
-    from nshtrainer.profiler import PyTorchProfilerConfig as PyTorchProfilerConfig
-    from nshtrainer.profiler import SimpleProfilerConfig as SimpleProfilerConfig
-else:
-    def __getattr__(name):
-        import importlib
-        if name in globals():
-            return globals()[name]
-        if name == "AdvancedProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").AdvancedProfilerConfig
-        if name == "BaseProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").BaseProfilerConfig
-        if name == "PyTorchProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").PyTorchProfilerConfig
-        if name == "SimpleProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").SimpleProfilerConfig
-        if name == "ProfilerConfig":
-            return importlib.import_module("nshtrainer.profiler").ProfilerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-# Submodule exports
 from . import _base as _base
 from . import advanced as advanced
 from . import pytorch as pytorch

nshtrainer/configs/profiler/_base/__init__.py CHANGED Viewed

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

nshtrainer/configs/profiler/advanced/__init__.py CHANGED Viewed

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

nshtrainer/configs/profiler/pytorch/__init__.py CHANGED Viewed

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

nshtrainer/configs/profiler/simple/__init__.py CHANGED Viewed

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

nshtrainer/configs/trainer/__init__.py CHANGED Viewed

@@ -2,175 +2,56 @@ from __future__ import annotations
 __codegen__ = True
-from typing import TYPE_CHECKING
+from nshtrainer.trainer import TrainerConfig as TrainerConfig
+from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
+from nshtrainer.trainer._config import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.trainer._config import (
+    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
+from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.trainer._config import (
+    CheckpointCallbackConfig as CheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
+from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
+from nshtrainer.trainer._config import (
+    DebugFlagCallbackConfig as DebugFlagCallbackConfig,
+)
+from nshtrainer.trainer._config import DirectoryConfig as DirectoryConfig
+from nshtrainer.trainer._config import (
+    EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+)
+from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
+from nshtrainer.trainer._config import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)
+from nshtrainer.trainer._config import LogEpochCallbackConfig as LogEpochCallbackConfig
+from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
+from nshtrainer.trainer._config import MetricConfig as MetricConfig
+from nshtrainer.trainer._config import (
+    NormLoggingCallbackConfig as NormLoggingCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+from nshtrainer.trainer._config import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)
+from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+from nshtrainer.trainer._config import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    TensorboardLoggerConfig as TensorboardLoggerConfig,
+)
+from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
-# Config/alias imports
-if TYPE_CHECKING:
-    from nshtrainer.trainer import TrainerConfig as TrainerConfig
-    from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
-    from nshtrainer.trainer._config import BaseLoggerConfig as BaseLoggerConfig
-    from nshtrainer.trainer._config import (
-        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
-    from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
-    from nshtrainer.trainer._config import (
-        CheckpointCallbackConfig as CheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        CheckpointSavingConfig as CheckpointSavingConfig,
-    )
-    from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
-    from nshtrainer.trainer._config import (
-        DebugFlagCallbackConfig as DebugFlagCallbackConfig,
-    )
-    from nshtrainer.trainer._config import DirectoryConfig as DirectoryConfig
-    from nshtrainer.trainer._config import (
-        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
-    )
-    from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
-    from nshtrainer.trainer._config import (
-        GradientClippingConfig as GradientClippingConfig,
-    )
-    from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
-    from nshtrainer.trainer._config import (
-        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        LearningRateMonitorConfig as LearningRateMonitorConfig,
-    )
-    from nshtrainer.trainer._config import (
-        LogEpochCallbackConfig as LogEpochCallbackConfig,
-    )
-    from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
-    from nshtrainer.trainer._config import MetricConfig as MetricConfig
-    from nshtrainer.trainer._config import (
-        NormLoggingCallbackConfig as NormLoggingCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
-    from nshtrainer.trainer._config import (
-        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
-    )
-    from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
-    from nshtrainer.trainer._config import (
-        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        TensorboardLoggerConfig as TensorboardLoggerConfig,
-    )
-    from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
-else:
-    def __getattr__(name):
-        import importlib
-        if name in globals():
-            return globals()[name]
-        if name == "ActSaveLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).ActSaveLoggerConfig
-        if name == "BaseLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).BaseLoggerConfig
-        if name == "BestCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).BestCheckpointCallbackConfig
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CallbackConfigBase
-        if name == "CheckpointSavingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CheckpointSavingConfig
-        if name == "DebugFlagCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).DebugFlagCallbackConfig
-        if name == "DirectoryConfig":
-            return importlib.import_module("nshtrainer.trainer._config").DirectoryConfig
-        if name == "EarlyStoppingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).EarlyStoppingCallbackConfig
-        if name == "EnvironmentConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).EnvironmentConfig
-        if name == "GradientClippingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).GradientClippingConfig
-        if name == "HuggingFaceHubConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).HuggingFaceHubConfig
-        if name == "LastCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LastCheckpointCallbackConfig
-        if name == "LearningRateMonitorConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LearningRateMonitorConfig
-        if name == "LogEpochCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LogEpochCallbackConfig
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.trainer._config").MetricConfig
-        if name == "NormLoggingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).NormLoggingCallbackConfig
-        if name == "OnExceptionCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).OnExceptionCheckpointCallbackConfig
-        if name == "RLPSanityChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).RLPSanityChecksCallbackConfig
-        if name == "SanityCheckingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).SanityCheckingConfig
-        if name == "SharedParametersCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).SharedParametersCallbackConfig
-        if name == "TensorboardLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).TensorboardLoggerConfig
-        if name == "TrainerConfig":
-            return importlib.import_module("nshtrainer.trainer").TrainerConfig
-        if name == "WandbLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).WandbLoggerConfig
-        if name == "CallbackConfig":
-            return importlib.import_module("nshtrainer.trainer._config").CallbackConfig
-        if name == "CheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CheckpointCallbackConfig
-        if name == "LoggerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").LoggerConfig
-        if name == "ProfilerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").ProfilerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-# Submodule exports
 from . import _config as _config
 from . import trainer as trainer

nshtrainer/configs/trainer/_config/__init__.py CHANGED Viewed

@@ -2,172 +2,53 @@ from __future__ import annotations
 __codegen__ = True
-from typing import TYPE_CHECKING
-# Config/alias imports
-if TYPE_CHECKING:
-    from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
-    from nshtrainer.trainer._config import BaseLoggerConfig as BaseLoggerConfig
-    from nshtrainer.trainer._config import (
-        BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
-    from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
-    from nshtrainer.trainer._config import (
-        CheckpointCallbackConfig as CheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        CheckpointSavingConfig as CheckpointSavingConfig,
-    )
-    from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
-    from nshtrainer.trainer._config import (
-        DebugFlagCallbackConfig as DebugFlagCallbackConfig,
-    )
-    from nshtrainer.trainer._config import DirectoryConfig as DirectoryConfig
-    from nshtrainer.trainer._config import (
-        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
-    )
-    from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
-    from nshtrainer.trainer._config import (
-        GradientClippingConfig as GradientClippingConfig,
-    )
-    from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
-    from nshtrainer.trainer._config import (
-        LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        LearningRateMonitorConfig as LearningRateMonitorConfig,
-    )
-    from nshtrainer.trainer._config import (
-        LogEpochCallbackConfig as LogEpochCallbackConfig,
-    )
-    from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
-    from nshtrainer.trainer._config import MetricConfig as MetricConfig
-    from nshtrainer.trainer._config import (
-        NormLoggingCallbackConfig as NormLoggingCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
-    )
-    from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
-    from nshtrainer.trainer._config import (
-        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
-    )
-    from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
-    from nshtrainer.trainer._config import (
-        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
-    )
-    from nshtrainer.trainer._config import (
-        TensorboardLoggerConfig as TensorboardLoggerConfig,
-    )
-    from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
-    from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig
-else:
-    def __getattr__(name):
-        import importlib
-        if name in globals():
-            return globals()[name]
-        if name == "ActSaveLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).ActSaveLoggerConfig
-        if name == "BaseLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).BaseLoggerConfig
-        if name == "BestCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).BestCheckpointCallbackConfig
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CallbackConfigBase
-        if name == "CheckpointSavingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CheckpointSavingConfig
-        if name == "DebugFlagCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).DebugFlagCallbackConfig
-        if name == "DirectoryConfig":
-            return importlib.import_module("nshtrainer.trainer._config").DirectoryConfig
-        if name == "EarlyStoppingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).EarlyStoppingCallbackConfig
-        if name == "EnvironmentConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).EnvironmentConfig
-        if name == "GradientClippingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).GradientClippingConfig
-        if name == "HuggingFaceHubConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).HuggingFaceHubConfig
-        if name == "LastCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LastCheckpointCallbackConfig
-        if name == "LearningRateMonitorConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LearningRateMonitorConfig
-        if name == "LogEpochCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).LogEpochCallbackConfig
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.trainer._config").MetricConfig
-        if name == "NormLoggingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).NormLoggingCallbackConfig
-        if name == "OnExceptionCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).OnExceptionCheckpointCallbackConfig
-        if name == "RLPSanityChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).RLPSanityChecksCallbackConfig
-        if name == "SanityCheckingConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).SanityCheckingConfig
-        if name == "SharedParametersCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).SharedParametersCallbackConfig
-        if name == "TensorboardLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).TensorboardLoggerConfig
-        if name == "TrainerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").TrainerConfig
-        if name == "WandbLoggerConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).WandbLoggerConfig
-        if name == "CallbackConfig":
-            return importlib.import_module("nshtrainer.trainer._config").CallbackConfig
-        if name == "CheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).CheckpointCallbackConfig
-        if name == "LoggerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").LoggerConfig
-        if name == "ProfilerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").ProfilerConfig
-        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
-# Submodule exports
+from nshtrainer.trainer._config import ActSaveLoggerConfig as ActSaveLoggerConfig
+from nshtrainer.trainer._config import BaseLoggerConfig as BaseLoggerConfig
+from nshtrainer.trainer._config import (
+    BestCheckpointCallbackConfig as BestCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import CallbackConfig as CallbackConfig
+from nshtrainer.trainer._config import CallbackConfigBase as CallbackConfigBase
+from nshtrainer.trainer._config import (
+    CheckpointCallbackConfig as CheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import CheckpointSavingConfig as CheckpointSavingConfig
+from nshtrainer.trainer._config import CSVLoggerConfig as CSVLoggerConfig
+from nshtrainer.trainer._config import (
+    DebugFlagCallbackConfig as DebugFlagCallbackConfig,
+)
+from nshtrainer.trainer._config import DirectoryConfig as DirectoryConfig
+from nshtrainer.trainer._config import (
+    EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+)
+from nshtrainer.trainer._config import EnvironmentConfig as EnvironmentConfig
+from nshtrainer.trainer._config import GradientClippingConfig as GradientClippingConfig
+from nshtrainer.trainer._config import HuggingFaceHubConfig as HuggingFaceHubConfig
+from nshtrainer.trainer._config import (
+    LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    LearningRateMonitorConfig as LearningRateMonitorConfig,
+)
+from nshtrainer.trainer._config import LogEpochCallbackConfig as LogEpochCallbackConfig
+from nshtrainer.trainer._config import LoggerConfig as LoggerConfig
+from nshtrainer.trainer._config import MetricConfig as MetricConfig
+from nshtrainer.trainer._config import (
+    NormLoggingCallbackConfig as NormLoggingCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    OnExceptionCheckpointCallbackConfig as OnExceptionCheckpointCallbackConfig,
+)
+from nshtrainer.trainer._config import ProfilerConfig as ProfilerConfig
+from nshtrainer.trainer._config import (
+    RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
+)
+from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
+from nshtrainer.trainer._config import (
+    SharedParametersCallbackConfig as SharedParametersCallbackConfig,
+)
+from nshtrainer.trainer._config import (
+    TensorboardLoggerConfig as TensorboardLoggerConfig,
+)
+from nshtrainer.trainer._config import TrainerConfig as TrainerConfig
+from nshtrainer.trainer._config import WandbLoggerConfig as WandbLoggerConfig

nshtrainer/configs/trainer/trainer/__init__.py CHANGED Viewed

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

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

nshtrainer 1.0.0b14py3-none-any.whl → 1.0.0b16py3-none-any.whl