nshtrainer 0.8.3__tar.gz → 0.8.5__tar.gz

{nshtrainer-0.8.3 → nshtrainer-0.8.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.8.3
+Version: 0.8.5
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshtrainer-0.8.3 → nshtrainer-0.8.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nshtrainer"
-version = "0.8.3"
+version = "0.8.5"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"

nshtrainer-0.8.5/src/nshtrainer/__init__.py

@@ -0,0 +1,14 @@
+from . import _experimental as _experimental
+from . import callbacks as callbacks
+from . import data as data
+from . import lr_scheduler as lr_scheduler
+from . import model as model
+from . import nn as nn
+from . import optimizer as optimizer
+from .model import Base as Base
+from .model import BaseConfig as BaseConfig
+from .model import ConfigList as ConfigList
+from .model import LightningModuleBase as LightningModuleBase
+from .model import MetricConfig as MetricConfig
+from .runner import Runner as Runner
+from .trainer import Trainer as Trainer

{nshtrainer-0.8.3 → nshtrainer-0.8.5}/src/nshtrainer/ll/__init__.py

@@ -1,6 +1,7 @@
 from . import _experimental as _experimental
 from . import actsave as actsave
 from . import callbacks as callbacks
+from . import data as data
 from . import lr_scheduler as lr_scheduler
 from . import model as model
 from . import nn as nn

{nshtrainer-0.8.3 → nshtrainer-0.8.5}/src/nshtrainer/model/base.py

@@ -15,7 +15,7 @@ from lightning.fabric.utilities.types import _MAP_LOCATION_TYPE, _PATH
 from lightning.pytorch import LightningDataModule, LightningModule, Trainer
 from lightning.pytorch.callbacks import Callback
 from lightning.pytorch.utilities.types import STEP_OUTPUT
-from typing_extensions import Self, TypeVar, deprecated, override
+from typing_extensions import Self, TypeVar, override
 
 from .config import (
     BaseConfig,
@@ -39,12 +39,6 @@ THparams = TypeVar("THparams", bound=BaseConfig, infer_variance=True)
 
 
 class Base(DebugModuleMixin, Generic[THparams]):
-    @deprecated("Use `ll.nn.MLP` instead.")
-    def mlp(self, *args, **kwargs):
-        from ..nn.mlp import MLP
-
-        return MLP(*args, **kwargs)
-
     @torch.jit.unused
     @property
     def config(self) -> THparams:

nshtrainer-0.8.3/src/nshtrainer/__init__.py

@@ -1,42 +0,0 @@
-from . import _experimental as _experimental
-from . import callbacks as callbacks
-from . import lr_scheduler as lr_scheduler
-from . import model as model
-from . import nn as nn
-from . import optimizer as optimizer
-from .data import dataset_transform as dataset_transform
-from .lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
-from .model import ActSaveConfig as ActSaveConfig
-from .model import Base as Base
-from .model import BaseConfig as BaseConfig
-from .model import BaseLoggerConfig as BaseLoggerConfig
-from .model import BaseProfilerConfig as BaseProfilerConfig
-from .model import CheckpointLoadingConfig as CheckpointLoadingConfig
-from .model import CheckpointSavingConfig as CheckpointSavingConfig
-from .model import ConfigList as ConfigList
-from .model import DirectoryConfig as DirectoryConfig
-from .model import (
-    EnvironmentClassInformationConfig as EnvironmentClassInformationConfig,
-)
-from .model import EnvironmentConfig as EnvironmentConfig
-from .model import (
-    EnvironmentLinuxEnvironmentConfig as EnvironmentLinuxEnvironmentConfig,
-)
-from .model import (
-    EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
-)
-from .model import GradientClippingConfig as GradientClippingConfig
-from .model import LightningModuleBase as LightningModuleBase
-from .model import LoggingConfig as LoggingConfig
-from .model import MetricConfig as MetricConfig
-from .model import OptimizationConfig as OptimizationConfig
-from .model import PrimaryMetricConfig as PrimaryMetricConfig
-from .model import ReproducibilityConfig as ReproducibilityConfig
-from .model import SanityCheckingConfig as SanityCheckingConfig
-from .model import TrainerConfig as TrainerConfig
-from .model import WandbWatchConfig as WandbWatchConfig
-from .nn import TypedModuleDict as TypedModuleDict
-from .nn import TypedModuleList as TypedModuleList
-from .optimizer import OptimizerConfig as OptimizerConfig
-from .runner import Runner as Runner
-from .trainer import Trainer as Trainer