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

nshtrainer/config/util/_environment_info/__init__.py

@@ -1,22 +1,94 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.util._environment_info import EnvironmentSnapshotConfig as EnvironmentSnapshotConfig
-from nshtrainer.util._environment_info import EnvironmentHardwareConfig as EnvironmentHardwareConfig
-from nshtrainer.util._environment_info import EnvironmentPackageConfig as EnvironmentPackageConfig
-from nshtrainer.util._environment_info import GitRepositoryConfig as GitRepositoryConfig
-from nshtrainer.util._environment_info import EnvironmentGPUConfig as EnvironmentGPUConfig
-from nshtrainer.util._environment_info import EnvironmentCUDAConfig as EnvironmentCUDAConfig
-from nshtrainer.util._environment_info import EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig
-from nshtrainer.util._environment_info import EnvironmentClassInformationConfig as EnvironmentClassInformationConfig
-from nshtrainer.util._environment_info import EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig
-from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
-from nshtrainer.util._environment_info import EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.util._environment_info import (
+        EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
+    )
+    from nshtrainer.util._environment_info import EnvironmentConfig as EnvironmentConfig
+    from nshtrainer.util._environment_info import (
+        EnvironmentCUDAConfig as EnvironmentCUDAConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentGPUConfig as EnvironmentGPUConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentHardwareConfig as EnvironmentHardwareConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentLSFInformationConfig as EnvironmentLSFInformationConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentPackageConfig as EnvironmentPackageConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        EnvironmentSnapshotConfig as EnvironmentSnapshotConfig,
+    )
+    from nshtrainer.util._environment_info import (
+        GitRepositoryConfig as GitRepositoryConfig,
+    )
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "EnvironmentLinuxEnvironmentConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentLinuxEnvironmentConfig
+        if name == "EnvironmentLSFInformationConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentLSFInformationConfig
+        if name == "EnvironmentGPUConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentGPUConfig
+        if name == "EnvironmentPackageConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentPackageConfig
+        if name == "EnvironmentHardwareConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentHardwareConfig
+        if name == "EnvironmentSnapshotConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentSnapshotConfig
+        if name == "EnvironmentClassInformationConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentClassInformationConfig
+        if name == "GitRepositoryConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).GitRepositoryConfig
+        if name == "EnvironmentConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentConfig
+        if name == "EnvironmentCUDAConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentCUDAConfig
+        if name == "EnvironmentSLURMInformationConfig":
+            return importlib.import_module(
+                "nshtrainer.util._environment_info"
+            ).EnvironmentSLURMInformationConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/config/util/config/__init__.py

@@ -1,16 +1,33 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.util.config import EpochsConfig as EpochsConfig
-from nshtrainer.util.config import StepsConfig as StepsConfig
-from nshtrainer.util.config import DTypeConfig as DTypeConfig
+from typing import TYPE_CHECKING
+
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.util.config import DTypeConfig as DTypeConfig
+    from nshtrainer.util.config import DurationConfig as DurationConfig
+    from nshtrainer.util.config import EpochsConfig as EpochsConfig
+    from nshtrainer.util.config import StepsConfig as StepsConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "EpochsConfig":
+            return importlib.import_module("nshtrainer.util.config").EpochsConfig
+        if name == "StepsConfig":
+            return importlib.import_module("nshtrainer.util.config").StepsConfig
+        if name == "DTypeConfig":
+            return importlib.import_module("nshtrainer.util.config").DTypeConfig
+        if name == "DurationConfig":
+            return importlib.import_module("nshtrainer.util.config").DurationConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
-# Type aliases
-from nshtrainer.util.config import DurationConfig as DurationConfig
 
 # Submodule exports
 from . import dtype as dtype

nshtrainer/config/util/config/dtype/__init__.py

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

nshtrainer/config/util/config/duration/__init__.py

@@ -1,14 +1,34 @@
-# fmt: off
-# ruff: noqa
-# type: ignore
+from __future__ import annotations
 
 __codegen__ = True
 
-# Config classes
-from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
-from nshtrainer.util.config.duration import StepsConfig as StepsConfig
+from typing import TYPE_CHECKING
 
-# Type aliases
-from nshtrainer.util.config.duration import DurationConfig as DurationConfig
+# Config/alias imports
+
+if TYPE_CHECKING:
+    from nshtrainer.util.config.duration import DurationConfig as DurationConfig
+    from nshtrainer.util.config.duration import EpochsConfig as EpochsConfig
+    from nshtrainer.util.config.duration import StepsConfig as StepsConfig
+else:
+
+    def __getattr__(name):
+        import importlib
+
+        if name in globals():
+            return globals()[name]
+        if name == "EpochsConfig":
+            return importlib.import_module(
+                "nshtrainer.util.config.duration"
+            ).EpochsConfig
+        if name == "StepsConfig":
+            return importlib.import_module(
+                "nshtrainer.util.config.duration"
+            ).StepsConfig
+        if name == "DurationConfig":
+            return importlib.import_module(
+                "nshtrainer.util.config.duration"
+            ).DurationConfig
+        raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
 
 # Submodule exports

nshtrainer/data/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from . import transform as dataset_transform
 from .balanced_batch_sampler import BalancedBatchSampler as BalancedBatchSampler
 from .datamodule import LightningDataModuleBase as LightningDataModuleBase

nshtrainer/data/balanced_batch_sampler.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import heapq
 import logging
 from typing import Any, Protocol, runtime_checkable

nshtrainer/data/datamodule.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from lightning.pytorch import LightningDataModule
 
 

nshtrainer/data/transform.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 from collections.abc import Callable
 from typing import Any, cast

nshtrainer/ll/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TypeAlias
 
 from . import _experimental as _experimental

nshtrainer/ll/_experimental.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer._experimental import *  # noqa: F403

nshtrainer/ll/actsave.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from nshutils.actsave import *  # type: ignore # noqa: F403
 
 from nshtrainer.callbacks.actsave import ActSaveCallback as ActSaveCallback

nshtrainer/ll/callbacks.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.callbacks import *  # noqa: F403

nshtrainer/ll/config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from nshconfig import *  # type: ignore  # noqa: F403
 from nshconfig import Config as TypedConfig  # type: ignore  # noqa: F401
 

nshtrainer/ll/data.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.data import *  # noqa: F403

nshtrainer/ll/log.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from nshutils import init_python_logging as init_python_logging
 from nshutils import lovely as lovely
 from nshutils import pretty as pretty

nshtrainer/ll/lr_scheduler.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.lr_scheduler import *  # noqa: F403

nshtrainer/ll/model.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from nshtrainer.model import *  # noqa: F403
 
 from ..trainer._config import CheckpointLoadingConfig as CheckpointLoadingConfig

nshtrainer/ll/nn.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.nn import *  # noqa: F403

nshtrainer/ll/optimizer.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.optimizer import *  # noqa: F403

nshtrainer/ll/runner.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from nshrunner import SnapshotConfig as SnapshotConfig
 
 from nshtrainer.runner import *  # type: ignore # noqa: F403

nshtrainer/ll/snapshot.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshsnap import *  # pyright: ignore[reportWildcardImportFromLibrary] # noqa: F403

nshtrainer/ll/snoop.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshutils import snoop as snoop

nshtrainer/ll/trainer.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.trainer import *  # noqa: F403

nshtrainer/ll/typecheck.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshutils.typecheck import *  # type: ignore # noqa: F403

nshtrainer/ll/util.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from nshtrainer.util import *  # noqa: F403

nshtrainer/loggers/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Annotated, TypeAlias
 
 import nshconfig as C

nshtrainer/loggers/_base.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING
 

nshtrainer/loggers/csv.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Literal
 
 from typing_extensions import override

nshtrainer/loggers/tensorboard.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Literal
 

nshtrainer/loggers/wandb.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import importlib.metadata
 import logging
 from typing import TYPE_CHECKING, Literal
@@ -8,8 +10,8 @@ from packaging import version
 from typing_extensions import assert_never, override
 
 from ..callbacks.base import CallbackConfigBase
-from ..callbacks.wandb_upload_code import WandbUploadCodeConfig
-from ..callbacks.wandb_watch import WandbWatchConfig
+from ..callbacks.wandb_upload_code import WandbUploadCodeCallbackConfig
+from ..callbacks.wandb_watch import WandbWatchCallbackConfig
 from ._base import BaseLoggerConfig
 
 if TYPE_CHECKING:
@@ -92,10 +94,10 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
     - "none" or False: Do not log any checkpoints
     """
 
-    log_code: WandbUploadCodeConfig | None = WandbUploadCodeConfig()
+    log_code: WandbUploadCodeCallbackConfig | None = WandbUploadCodeCallbackConfig()
     """WandB code upload configuration. Used to upload code to WandB."""
 
-    watch: WandbWatchConfig | None = WandbWatchConfig()
+    watch: WandbWatchCallbackConfig | None = WandbWatchCallbackConfig()
     """WandB model watch configuration. Used to log model architecture, gradients, and parameters."""
 
     offline: bool = False

nshtrainer/lr_scheduler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Annotated, TypeAlias
 
 import nshconfig as C

nshtrainer/lr_scheduler/_base.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import math
 from abc import ABC, abstractmethod
 from collections.abc import Mapping

nshtrainer/lr_scheduler/linear_warmup_cosine.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import math
 import warnings
 from typing import Literal

nshtrainer/lr_scheduler/reduce_lr_on_plateau.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TYPE_CHECKING, Literal, cast
 
 from lightning.pytorch.utilities.types import LRSchedulerConfigType

nshtrainer/metrics/__init__.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from ._config import MetricConfig as MetricConfig

nshtrainer/metrics/_config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import builtins
 from typing import Literal
 

nshtrainer/model/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from .base import LightningModuleBase as LightningModuleBase
 from .config import BaseConfig as BaseConfig
 from .config import DirectoryConfig as DirectoryConfig

nshtrainer/model/base.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import inspect
 import logging
 from abc import ABC, abstractmethod

nshtrainer/model/config.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 import logging
 import os

nshtrainer/model/mixins/callback.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from collections.abc import Callable, Iterable, Sequence
 from typing import Any, TypeAlias, cast, final

nshtrainer/model/mixins/logger.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections import deque
 from collections.abc import Callable, Generator
 from contextlib import contextmanager

nshtrainer/nn/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from .mlp import MLP as MLP
 from .mlp import MLPConfig as MLPConfig
 from .mlp import MLPConfigDict as MLPConfigDict

nshtrainer/nn/mlp.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 from collections.abc import Callable, Sequence
 from typing import Literal, Protocol, runtime_checkable

nshtrainer/nn/module_dict.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Iterable, Mapping
 from typing import Generic, cast
 

nshtrainer/nn/module_list.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Iterable, Iterator
 from typing import Generic, TypeVar, overload
 

nshtrainer/nn/nonlinearity.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import Annotated, Literal
 

nshtrainer/optimizer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from collections.abc import Iterable
 from typing import Annotated, Any, Literal, TypeAlias

nshtrainer/profiler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Annotated, TypeAlias
 
 import nshconfig as C

nshtrainer/profiler/_base.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from abc import ABC, abstractmethod
 from pathlib import Path

nshtrainer/profiler/advanced.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Literal
 

nshtrainer/profiler/pytorch.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Any, Literal
 

nshtrainer/profiler/simple.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Literal
 

nshtrainer/runner.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 import logging
 from collections.abc import Callable, Iterable, Mapping, Sequence

nshtrainer/scripts/find_packages.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import argparse
 import ast
 import glob

nshtrainer/trainer/__init__.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 from .trainer import Trainer as Trainer