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

nshtrainer/config/loggers/csv/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/loggers/tensorboard/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/loggers/wandb/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -8,8 +10,12 @@ if TYPE_CHECKING:
     from nshtrainer.loggers.wandb import BaseLoggerConfig as BaseLoggerConfig
     from nshtrainer.loggers.wandb import CallbackConfigBase as CallbackConfigBase
     from nshtrainer.loggers.wandb import WandbLoggerConfig as WandbLoggerConfig
-    from nshtrainer.loggers.wandb import WandbUploadCodeConfig as WandbUploadCodeConfig
-    from nshtrainer.loggers.wandb import WandbWatchConfig as WandbWatchConfig
+    from nshtrainer.loggers.wandb import (
+        WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
+    )
+    from nshtrainer.loggers.wandb import (
+        WandbWatchCallbackConfig as WandbWatchCallbackConfig,
+    )
 else:
 
     def __getattr__(name):
@@ -17,20 +23,22 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module(
-                "nshtrainer.loggers.wandb"
-            ).CallbackConfigBase
         if name == "WandbLoggerConfig":
             return importlib.import_module("nshtrainer.loggers.wandb").WandbLoggerConfig
-        if name == "WandbUploadCodeConfig":
+        if name == "WandbUploadCodeCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.loggers.wandb"
-            ).WandbUploadCodeConfig
-        if name == "WandbWatchConfig":
-            return importlib.import_module("nshtrainer.loggers.wandb").WandbWatchConfig
+            ).WandbUploadCodeCallbackConfig
+        if name == "WandbWatchCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).WandbWatchCallbackConfig
         if name == "BaseLoggerConfig":
             return importlib.import_module("nshtrainer.loggers.wandb").BaseLoggerConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module(
+                "nshtrainer.loggers.wandb"
+            ).CallbackConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/lr_scheduler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/lr_scheduler/_base/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/lr_scheduler/linear_warmup_cosine/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/lr_scheduler/reduce_lr_on_plateau/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -21,6 +23,10 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "LRSchedulerConfigBase":
+            return importlib.import_module(
+                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
+            ).LRSchedulerConfigBase
         if name == "MetricConfig":
             return importlib.import_module(
                 "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
@@ -29,10 +35,6 @@ else:
             return importlib.import_module(
                 "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
             ).ReduceLROnPlateauConfig
-        if name == "LRSchedulerConfigBase":
-            return importlib.import_module(
-                "nshtrainer.lr_scheduler.reduce_lr_on_plateau"
-            ).LRSchedulerConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/metrics/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/metrics/_config/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/model/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -18,18 +20,18 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module("nshtrainer.model.config").CallbackConfigBase
+        if name == "MetricConfig":
+            return importlib.import_module("nshtrainer.model").MetricConfig
         if name == "TrainerConfig":
             return importlib.import_module("nshtrainer.model").TrainerConfig
-        if name == "EnvironmentConfig":
-            return importlib.import_module("nshtrainer.model.base").EnvironmentConfig
         if name == "BaseConfig":
             return importlib.import_module("nshtrainer.model").BaseConfig
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.model").MetricConfig
+        if name == "EnvironmentConfig":
+            return importlib.import_module("nshtrainer.model.base").EnvironmentConfig
         if name == "DirectoryConfig":
             return importlib.import_module("nshtrainer.model").DirectoryConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer.model.config").CallbackConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 

nshtrainer/config/model/base/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -14,10 +16,10 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "EnvironmentConfig":
-            return importlib.import_module("nshtrainer.model.base").EnvironmentConfig
         if name == "BaseConfig":
             return importlib.import_module("nshtrainer.model.base").BaseConfig
+        if name == "EnvironmentConfig":
+            return importlib.import_module("nshtrainer.model.base").EnvironmentConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/model/config/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -18,18 +20,18 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CallbackConfigBase":
-            return importlib.import_module("nshtrainer.model.config").CallbackConfigBase
+        if name == "MetricConfig":
+            return importlib.import_module("nshtrainer.model.config").MetricConfig
         if name == "TrainerConfig":
             return importlib.import_module("nshtrainer.model.config").TrainerConfig
-        if name == "EnvironmentConfig":
-            return importlib.import_module("nshtrainer.model.config").EnvironmentConfig
         if name == "BaseConfig":
             return importlib.import_module("nshtrainer.model.config").BaseConfig
-        if name == "MetricConfig":
-            return importlib.import_module("nshtrainer.model.config").MetricConfig
+        if name == "EnvironmentConfig":
+            return importlib.import_module("nshtrainer.model.config").EnvironmentConfig
         if name == "DirectoryConfig":
             return importlib.import_module("nshtrainer.model.config").DirectoryConfig
+        if name == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer.model.config").CallbackConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/model/mixins/logger/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/nn/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -35,36 +37,36 @@ else:
             return importlib.import_module("nshtrainer.nn").BaseNonlinearityConfig
         if name == "MLPConfig":
             return importlib.import_module("nshtrainer.nn").MLPConfig
+        if name == "PReLUConfig":
+            return importlib.import_module("nshtrainer.nn").PReLUConfig
+        if name == "LeakyReLUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").LeakyReLUNonlinearityConfig
         if name == "SwiGLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SwiGLUNonlinearityConfig
-        if name == "ReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").ReLUNonlinearityConfig
+        if name == "SoftsignNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SoftsignNonlinearityConfig
         if name == "SiLUNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SiLUNonlinearityConfig
+        if name == "SigmoidNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SigmoidNonlinearityConfig
+        if name == "SoftplusNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SoftplusNonlinearityConfig
         if name == "ELUNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").ELUNonlinearityConfig
+        if name == "SoftmaxNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").SoftmaxNonlinearityConfig
         if name == "GELUNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").GELUNonlinearityConfig
-        if name == "SoftplusNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftplusNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftsignNonlinearityConfig
         if name == "SwishNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").SwishNonlinearityConfig
-        if name == "SoftmaxNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SoftmaxNonlinearityConfig
         if name == "MishNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").MishNonlinearityConfig
-        if name == "SigmoidNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").SigmoidNonlinearityConfig
         if name == "TanhNonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").TanhNonlinearityConfig
-        if name == "PReLUConfig":
-            return importlib.import_module("nshtrainer.nn").PReLUConfig
-        if name == "LeakyReLUNonlinearityConfig":
-            return importlib.import_module("nshtrainer.nn").LeakyReLUNonlinearityConfig
+        if name == "ReLUNonlinearityConfig":
+            return importlib.import_module("nshtrainer.nn").ReLUNonlinearityConfig
         if name == "NonlinearityConfig":
             return importlib.import_module("nshtrainer.nn").NonlinearityConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

nshtrainer/config/nn/mlp/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/nn/nonlinearity/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -56,64 +58,64 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "PReLUConfig":
+            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
+        if name == "LeakyReLUNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).LeakyReLUNonlinearityConfig
         if name == "SwiGLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SwiGLUNonlinearityConfig
-        if name == "ReLUNonlinearityConfig":
+        if name == "SoftsignNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).ReLUNonlinearityConfig
+            ).SoftsignNonlinearityConfig
         if name == "SiLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SiLUNonlinearityConfig
-        if name == "ELUNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).ELUNonlinearityConfig
-        if name == "GELUNonlinearityConfig":
+        if name == "SigmoidNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).GELUNonlinearityConfig
+            ).SigmoidNonlinearityConfig
         if name == "SoftplusNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SoftplusNonlinearityConfig
-        if name == "SoftsignNonlinearityConfig":
-            return importlib.import_module(
-                "nshtrainer.nn.nonlinearity"
-            ).SoftsignNonlinearityConfig
-        if name == "SwishNonlinearityConfig":
+        if name == "ELUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SwishNonlinearityConfig
+            ).ELUNonlinearityConfig
         if name == "SoftmaxNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).SoftmaxNonlinearityConfig
-        if name == "MishNonlinearityConfig":
+        if name == "GELUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).MishNonlinearityConfig
-        if name == "SigmoidNonlinearityConfig":
+            ).GELUNonlinearityConfig
+        if name == "SwishNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).SigmoidNonlinearityConfig
+            ).SwishNonlinearityConfig
+        if name == "MishNonlinearityConfig":
+            return importlib.import_module(
+                "nshtrainer.nn.nonlinearity"
+            ).MishNonlinearityConfig
         if name == "TanhNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
             ).TanhNonlinearityConfig
-        if name == "BaseNonlinearityConfig":
+        if name == "ReLUNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).BaseNonlinearityConfig
-        if name == "PReLUConfig":
-            return importlib.import_module("nshtrainer.nn.nonlinearity").PReLUConfig
-        if name == "LeakyReLUNonlinearityConfig":
+            ).ReLUNonlinearityConfig
+        if name == "BaseNonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"
-            ).LeakyReLUNonlinearityConfig
+            ).BaseNonlinearityConfig
         if name == "NonlinearityConfig":
             return importlib.import_module(
                 "nshtrainer.nn.nonlinearity"

nshtrainer/config/optimizer/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/profiler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/profiler/_base/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/profiler/advanced/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -16,14 +18,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "BaseProfilerConfig":
-            return importlib.import_module(
-                "nshtrainer.profiler.advanced"
-            ).BaseProfilerConfig
         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

nshtrainer/config/profiler/pytorch/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -16,14 +18,14 @@ else:
 
         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
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.pytorch"
+            ).BaseProfilerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/profiler/simple/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -14,14 +16,14 @@ else:
 
         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
+        if name == "BaseProfilerConfig":
+            return importlib.import_module(
+                "nshtrainer.profiler.simple"
+            ).BaseProfilerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/runner/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING

nshtrainer/config/trainer/_config/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING
@@ -23,7 +25,9 @@ if TYPE_CHECKING:
     from nshtrainer.trainer._config import (
         DebugFlagCallbackConfig as DebugFlagCallbackConfig,
     )
-    from nshtrainer.trainer._config import EarlyStoppingConfig as EarlyStoppingConfig
+    from nshtrainer.trainer._config import (
+        EarlyStoppingCallbackConfig as EarlyStoppingCallbackConfig,
+    )
     from nshtrainer.trainer._config import (
         GradientClippingConfig as GradientClippingConfig,
     )
@@ -42,11 +46,11 @@ if TYPE_CHECKING:
         ReproducibilityConfig as ReproducibilityConfig,
     )
     from nshtrainer.trainer._config import (
-        RLPSanityChecksConfig as RLPSanityChecksConfig,
+        RLPSanityChecksCallbackConfig as RLPSanityChecksCallbackConfig,
     )
     from nshtrainer.trainer._config import SanityCheckingConfig as SanityCheckingConfig
     from nshtrainer.trainer._config import (
-        SharedParametersConfig as SharedParametersConfig,
+        SharedParametersCallbackConfig as SharedParametersCallbackConfig,
     )
     from nshtrainer.trainer._config import (
         TensorboardLoggerConfig as TensorboardLoggerConfig,
@@ -60,80 +64,80 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "HuggingFaceHubConfig":
-            return importlib.import_module(
-                "nshtrainer.trainer._config"
-            ).HuggingFaceHubConfig
-        if name == "OptimizationConfig":
+        if name == "SanityCheckingConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).OptimizationConfig
+            ).SanityCheckingConfig
         if name == "TrainerConfig":
             return importlib.import_module("nshtrainer.trainer._config").TrainerConfig
-        if name == "TensorboardLoggerConfig":
+        if name == "OnExceptionCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).TensorboardLoggerConfig
+            ).OnExceptionCheckpointCallbackConfig
         if name == "GradientClippingConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
             ).GradientClippingConfig
-        if name == "CallbackConfigBase":
+        if name == "WandbLoggerConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).CallbackConfigBase
-        if name == "CSVLoggerConfig":
-            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
-        if name == "LastCheckpointCallbackConfig":
+            ).WandbLoggerConfig
+        if name == "LoggingConfig":
+            return importlib.import_module("nshtrainer.trainer._config").LoggingConfig
+        if name == "TensorboardLoggerConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).LastCheckpointCallbackConfig
-        if name == "OnExceptionCheckpointCallbackConfig":
+            ).TensorboardLoggerConfig
+        if name == "RLPSanityChecksCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).OnExceptionCheckpointCallbackConfig
-        if name == "RLPSanityChecksConfig":
+            ).RLPSanityChecksCallbackConfig
+        if name == "CheckpointSavingConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).RLPSanityChecksConfig
-        if name == "EarlyStoppingConfig":
+            ).CheckpointSavingConfig
+        if name == "CSVLoggerConfig":
+            return importlib.import_module("nshtrainer.trainer._config").CSVLoggerConfig
+        if name == "HuggingFaceHubConfig":
+            return importlib.import_module(
+                "nshtrainer.trainer._config"
+            ).HuggingFaceHubConfig
+        if name == "CheckpointLoadingConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).EarlyStoppingConfig
+            ).CheckpointLoadingConfig
         if name == "DebugFlagCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
             ).DebugFlagCallbackConfig
-        if name == "WandbLoggerConfig":
+        if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).WandbLoggerConfig
-        if name == "CheckpointSavingConfig":
+            ).CallbackConfigBase
+        if name == "LastCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).CheckpointSavingConfig
-        if name == "CheckpointLoadingConfig":
+            ).LastCheckpointCallbackConfig
+        if name == "SharedParametersCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).CheckpointLoadingConfig
-        if name == "BestCheckpointCallbackConfig":
+            ).SharedParametersCallbackConfig
+        if name == "ReproducibilityConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).BestCheckpointCallbackConfig
-        if name == "LoggingConfig":
-            return importlib.import_module("nshtrainer.trainer._config").LoggingConfig
-        if name == "SanityCheckingConfig":
+            ).ReproducibilityConfig
+        if name == "EarlyStoppingCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).SanityCheckingConfig
-        if name == "SharedParametersConfig":
+            ).EarlyStoppingCallbackConfig
+        if name == "OptimizationConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).SharedParametersConfig
-        if name == "ReproducibilityConfig":
+            ).OptimizationConfig
+        if name == "BestCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.trainer._config"
-            ).ReproducibilityConfig
+            ).BestCheckpointCallbackConfig
         if name == "CallbackConfig":
             return importlib.import_module("nshtrainer.trainer._config").CallbackConfig
         if name == "CheckpointCallbackConfig":

nshtrainer/config/trainer/checkpoint_connector/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 __codegen__ = True
 
 from typing import TYPE_CHECKING