nshtrainer 0.44.0__py3-none-any.whl → 1.0.0b9__py3-none-any.whl

nshtrainer/config/_directory/__init__.py

@@ -19,12 +19,12 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "DirectoryConfig":
+            return importlib.import_module("nshtrainer._directory").DirectoryConfig
         if name == "DirectorySetupCallbackConfig":
             return importlib.import_module(
                 "nshtrainer._directory"
             ).DirectorySetupCallbackConfig
-        if name == "DirectoryConfig":
-            return importlib.import_module("nshtrainer._directory").DirectoryConfig
         if name == "LoggerConfig":
             return importlib.import_module("nshtrainer._directory").LoggerConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

nshtrainer/config/_hf_hub/__init__.py

@@ -19,14 +19,14 @@ else:
 
         if name in globals():
             return globals()[name]
+        if name == "CallbackConfigBase":
+            return importlib.import_module("nshtrainer._hf_hub").CallbackConfigBase
         if name == "HuggingFaceHubAutoCreateConfig":
             return importlib.import_module(
                 "nshtrainer._hf_hub"
             ).HuggingFaceHubAutoCreateConfig
         if name == "HuggingFaceHubConfig":
             return importlib.import_module("nshtrainer._hf_hub").HuggingFaceHubConfig
-        if name == "CallbackConfigBase":
-            return importlib.import_module("nshtrainer._hf_hub").CallbackConfigBase
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/__init__.py

@@ -32,6 +32,7 @@ if TYPE_CHECKING:
     from nshtrainer.callbacks import (
         LastCheckpointCallbackConfig as LastCheckpointCallbackConfig,
     )
+    from nshtrainer.callbacks import LogEpochCallbackConfig as LogEpochCallbackConfig
     from nshtrainer.callbacks import (
         NormLoggingCallbackConfig as NormLoggingCallbackConfig,
     )
@@ -47,7 +48,6 @@ if TYPE_CHECKING:
     from nshtrainer.callbacks import (
         SharedParametersCallbackConfig as SharedParametersCallbackConfig,
     )
-    from nshtrainer.callbacks import ThroughputMonitorConfig as ThroughputMonitorConfig
     from nshtrainer.callbacks import (
         WandbUploadCodeCallbackConfig as WandbUploadCodeCallbackConfig,
     )
@@ -69,88 +69,88 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "PrintTableMetricsCallbackConfig":
+        if name == "ActSaveConfig":
+            return importlib.import_module("nshtrainer.callbacks.actsave").ActSaveConfig
+        if name == "BaseCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).BaseCheckpointCallbackConfig
+        if name == "BestCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).PrintTableMetricsCallbackConfig
+            ).BestCheckpointCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module("nshtrainer.callbacks").CallbackConfigBase
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
         if name == "DebugFlagCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
             ).DebugFlagCallbackConfig
-        if name == "ThroughputMonitorConfig":
+        if name == "DirectorySetupCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).ThroughputMonitorConfig
-        if name == "GradientSkippingCallbackConfig":
+            ).DirectorySetupCallbackConfig
+        if name == "EMACallbackConfig":
+            return importlib.import_module("nshtrainer.callbacks").EMACallbackConfig
+        if name == "EarlyStoppingCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).GradientSkippingCallbackConfig
-        if name == "RLPSanityChecksCallbackConfig":
+            ).EarlyStoppingCallbackConfig
+        if name == "EpochTimerCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).RLPSanityChecksCallbackConfig
-        if name == "WandbUploadCodeCallbackConfig":
+            ).EpochTimerCallbackConfig
+        if name == "FiniteChecksCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).WandbUploadCodeCallbackConfig
-        if name == "MetricConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.early_stopping"
-            ).MetricConfig
-        if name == "EarlyStoppingCallbackConfig":
+            ).FiniteChecksCallbackConfig
+        if name == "GradientSkippingCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).EarlyStoppingCallbackConfig
-        if name == "WandbWatchCallbackConfig":
+            ).GradientSkippingCallbackConfig
+        if name == "LastCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).WandbWatchCallbackConfig
-        if name == "EMACallbackConfig":
-            return importlib.import_module("nshtrainer.callbacks").EMACallbackConfig
-        if name == "DirectorySetupCallbackConfig":
+            ).LastCheckpointCallbackConfig
+        if name == "LogEpochCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).DirectorySetupCallbackConfig
-        if name == "ActSaveConfig":
-            return importlib.import_module("nshtrainer.callbacks.actsave").ActSaveConfig
-        if name == "FiniteChecksCallbackConfig":
+            ).LogEpochCallbackConfig
+        if name == "MetricConfig":
             return importlib.import_module(
-                "nshtrainer.callbacks"
-            ).FiniteChecksCallbackConfig
+                "nshtrainer.callbacks.early_stopping"
+            ).MetricConfig
         if name == "NormLoggingCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
             ).NormLoggingCallbackConfig
-        if name == "EpochTimerCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks"
-            ).EpochTimerCallbackConfig
         if name == "OnExceptionCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
             ).OnExceptionCheckpointCallbackConfig
-        if name == "LastCheckpointCallbackConfig":
+        if name == "PrintTableMetricsCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).LastCheckpointCallbackConfig
+            ).PrintTableMetricsCallbackConfig
+        if name == "RLPSanityChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks"
+            ).RLPSanityChecksCallbackConfig
         if name == "SharedParametersCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
             ).SharedParametersCallbackConfig
-        if name == "BestCheckpointCallbackConfig":
+        if name == "WandbUploadCodeCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks"
-            ).BestCheckpointCallbackConfig
-        if name == "CheckpointMetadata":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint._base"
-            ).CheckpointMetadata
-        if name == "BaseCheckpointCallbackConfig":
+            ).WandbUploadCodeCallbackConfig
+        if name == "WandbWatchCallbackConfig":
             return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint._base"
-            ).BaseCheckpointCallbackConfig
+                "nshtrainer.callbacks"
+            ).WandbWatchCallbackConfig
         if name == "CallbackConfig":
             return importlib.import_module("nshtrainer.callbacks").CallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
@@ -166,11 +166,11 @@ from . import early_stopping as early_stopping
 from . import ema as ema
 from . import finite_checks as finite_checks
 from . import gradient_skipping as gradient_skipping
+from . import log_epoch as log_epoch
 from . import norm_logging as norm_logging
 from . import print_table as print_table
 from . import rlp_sanity_checks as rlp_sanity_checks
 from . import shared_parameters as shared_parameters
-from . import throughput_monitor as throughput_monitor
 from . import timer as timer
 from . import wandb_upload_code as wandb_upload_code
 from . import wandb_watch as wandb_watch

nshtrainer/config/callbacks/checkpoint/__init__.py

@@ -35,34 +35,34 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CheckpointMetadata":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint._base"
-            ).CheckpointMetadata
         if name == "BaseCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint._base"
             ).BaseCheckpointCallbackConfig
-        if name == "LastCheckpointCallbackConfig":
+        if name == "BestCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint"
-            ).LastCheckpointCallbackConfig
+            ).BestCheckpointCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint._base"
             ).CallbackConfigBase
-        if name == "OnExceptionCheckpointCallbackConfig":
+        if name == "CheckpointMetadata":
             return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint"
-            ).OnExceptionCheckpointCallbackConfig
-        if name == "BestCheckpointCallbackConfig":
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
+        if name == "LastCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint"
-            ).BestCheckpointCallbackConfig
+            ).LastCheckpointCallbackConfig
         if name == "MetricConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.best_checkpoint"
             ).MetricConfig
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint"
+            ).OnExceptionCheckpointCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 

nshtrainer/config/callbacks/checkpoint/_base/__init__.py

@@ -23,10 +23,6 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CheckpointMetadata":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint._base"
-            ).CheckpointMetadata
         if name == "BaseCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint._base"
@@ -35,6 +31,10 @@ else:
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint._base"
             ).CallbackConfigBase
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint._base"
+            ).CheckpointMetadata
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/checkpoint/best_checkpoint/__init__.py

@@ -26,22 +26,22 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CheckpointMetadata":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint.best_checkpoint"
-            ).CheckpointMetadata
         if name == "BaseCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.best_checkpoint"
             ).BaseCheckpointCallbackConfig
-        if name == "MetricConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint.best_checkpoint"
-            ).MetricConfig
         if name == "BestCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.best_checkpoint"
             ).BestCheckpointCallbackConfig
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.best_checkpoint"
+            ).CheckpointMetadata
+        if name == "MetricConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.best_checkpoint"
+            ).MetricConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/checkpoint/last_checkpoint/__init__.py

@@ -23,14 +23,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "CheckpointMetadata":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint.last_checkpoint"
-            ).CheckpointMetadata
         if name == "BaseCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.last_checkpoint"
             ).BaseCheckpointCallbackConfig
+        if name == "CheckpointMetadata":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.last_checkpoint"
+            ).CheckpointMetadata
         if name == "LastCheckpointCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.last_checkpoint"

nshtrainer/config/callbacks/checkpoint/on_exception_checkpoint/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "OnExceptionCheckpointCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.checkpoint.on_exception_checkpoint"
-            ).OnExceptionCheckpointCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.checkpoint.on_exception_checkpoint"
             ).CallbackConfigBase
+        if name == "OnExceptionCheckpointCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.checkpoint.on_exception_checkpoint"
+            ).OnExceptionCheckpointCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/debug_flag/__init__.py

@@ -18,14 +18,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "DebugFlagCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.debug_flag"
-            ).DebugFlagCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.debug_flag"
             ).CallbackConfigBase
+        if name == "DebugFlagCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.debug_flag"
+            ).DebugFlagCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/directory_setup/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "DirectorySetupCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.directory_setup"
-            ).DirectorySetupCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.directory_setup"
             ).CallbackConfigBase
+        if name == "DirectorySetupCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.directory_setup"
+            ).DirectorySetupCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/early_stopping/__init__.py

@@ -21,18 +21,18 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "MetricConfig":
+        if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.early_stopping"
-            ).MetricConfig
+            ).CallbackConfigBase
         if name == "EarlyStoppingCallbackConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.early_stopping"
             ).EarlyStoppingCallbackConfig
-        if name == "CallbackConfigBase":
+        if name == "MetricConfig":
             return importlib.import_module(
                 "nshtrainer.callbacks.early_stopping"
-            ).CallbackConfigBase
+            ).MetricConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/ema/__init__.py

@@ -16,12 +16,12 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "EMACallbackConfig":
-            return importlib.import_module("nshtrainer.callbacks.ema").EMACallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.ema"
             ).CallbackConfigBase
+        if name == "EMACallbackConfig":
+            return importlib.import_module("nshtrainer.callbacks.ema").EMACallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/finite_checks/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "FiniteChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.finite_checks"
-            ).FiniteChecksCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.finite_checks"
             ).CallbackConfigBase
+        if name == "FiniteChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.finite_checks"
+            ).FiniteChecksCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/gradient_skipping/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "GradientSkippingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.gradient_skipping"
-            ).GradientSkippingCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.gradient_skipping"
             ).CallbackConfigBase
+        if name == "GradientSkippingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.gradient_skipping"
+            ).GradientSkippingCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/{throughput_monitor → log_epoch}/__init__.py

@@ -7,11 +7,9 @@ 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,
+    from nshtrainer.callbacks.log_epoch import CallbackConfigBase as CallbackConfigBase
+    from nshtrainer.callbacks.log_epoch import (
+        LogEpochCallbackConfig as LogEpochCallbackConfig,
     )
 else:
 
@@ -20,14 +18,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "ThroughputMonitorConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.throughput_monitor"
-            ).ThroughputMonitorConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
-                "nshtrainer.callbacks.throughput_monitor"
+                "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

nshtrainer/config/callbacks/norm_logging/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "NormLoggingCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.norm_logging"
-            ).NormLoggingCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.norm_logging"
             ).CallbackConfigBase
+        if name == "NormLoggingCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.norm_logging"
+            ).NormLoggingCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/print_table/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "PrintTableMetricsCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.print_table"
-            ).PrintTableMetricsCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.print_table"
             ).CallbackConfigBase
+        if name == "PrintTableMetricsCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.print_table"
+            ).PrintTableMetricsCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/rlp_sanity_checks/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "RLPSanityChecksCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.rlp_sanity_checks"
-            ).RLPSanityChecksCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.rlp_sanity_checks"
             ).CallbackConfigBase
+        if name == "RLPSanityChecksCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.rlp_sanity_checks"
+            ).RLPSanityChecksCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/shared_parameters/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "SharedParametersCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.shared_parameters"
-            ).SharedParametersCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.shared_parameters"
             ).CallbackConfigBase
+        if name == "SharedParametersCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.shared_parameters"
+            ).SharedParametersCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/timer/__init__.py

@@ -18,14 +18,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "EpochTimerCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.timer"
-            ).EpochTimerCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.timer"
             ).CallbackConfigBase
+        if name == "EpochTimerCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.timer"
+            ).EpochTimerCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/wandb_upload_code/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "WandbUploadCodeCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_upload_code"
-            ).WandbUploadCodeCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.wandb_upload_code"
             ).CallbackConfigBase
+        if name == "WandbUploadCodeCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.wandb_upload_code"
+            ).WandbUploadCodeCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/callbacks/wandb_watch/__init__.py

@@ -20,14 +20,14 @@ else:
 
         if name in globals():
             return globals()[name]
-        if name == "WandbWatchCallbackConfig":
-            return importlib.import_module(
-                "nshtrainer.callbacks.wandb_watch"
-            ).WandbWatchCallbackConfig
         if name == "CallbackConfigBase":
             return importlib.import_module(
                 "nshtrainer.callbacks.wandb_watch"
             ).CallbackConfigBase
+        if name == "WandbWatchCallbackConfig":
+            return importlib.import_module(
+                "nshtrainer.callbacks.wandb_watch"
+            ).WandbWatchCallbackConfig
         raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports