nshtrainer 0.8.3__py3-none-any.whl → 0.8.5__py3-none-any.whl

nshtrainer/__init__.py

@@ -1,42 +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 .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

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/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.dist-info → nshtrainer-0.8.5.dist-info}/METADATA

@@ -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.dist-info → nshtrainer-0.8.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-nshtrainer/__init__.py,sha256=gWRJYlPoS0Xlyy1I0hob2zKxuqDB2fO4cfAryb--Prk,1992
+nshtrainer/__init__.py,sha256=nbZHdfTk0oWqsJgrSzdgk2DSf4CGhdZn79esoJGauO8,548
 nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
 nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
 nshtrainer/_experimental/flops/flop_counter.py,sha256=-sL0Fy6poXa__hyzUMdZScjPULp4coQELQpPU6p6dXU,25736
@@ -24,7 +24,7 @@ nshtrainer/callbacks/wandb_watch.py,sha256=pUpMsNxd03ex1rzOmFw2HzGOXjnQGaH84m8cc
 nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
 nshtrainer/data/balanced_batch_sampler.py,sha256=bcJBcQjh1hB1yKF_xSlT9AtEWv0BJjYc1CuH2BF-ea8,4392
 nshtrainer/data/transform.py,sha256=JeGxvytQly8hougrsdMmKG8gJ6qvFPDglJCO4Tp6STk,1795
-nshtrainer/ll/__init__.py,sha256=qraZm_bAvoLR69w5iiAic9-TT4sR0g_iKnttiXbb5y0,2658
+nshtrainer/ll/__init__.py,sha256=nxYPtoFOFAvzkD6O3EIuwCiRi_LedYa_EH-RIfDG91s,2685
 nshtrainer/ll/_experimental.py,sha256=oBQCKOEVYoxuUU9eLb-Fg2B2mzZD7SA0zfAO6lmWZ88,53
 nshtrainer/ll/actsave.py,sha256=QJ7yJIqvabpZzumX7PLPzkh6dfqY-zxiEdzv48VtZEY,123
 nshtrainer/ll/callbacks.py,sha256=AxyUmc8aGRSjx6WwwgXYCmdJ73rwLuEAEH0AGRosojQ,49
@@ -46,7 +46,7 @@ nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1T
 nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=mn6cyizyI_stkXtg6zxIEGF9btIxMRWigUHUTlUYCSw,5221
 nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-WKArwxxf7ZQizhmbCw,2782
 nshtrainer/model/__init__.py,sha256=y32Hla-5whpzLL2BtCJpBakSp8o-1nQbpO0j_-xq_Po,1864
-nshtrainer/model/base.py,sha256=k2Qk_rH4FsBOib2PISXnG97SBlgeD1w_jw02NE4zEQI,21460
+nshtrainer/model/base.py,sha256=YtqnjiMf0cLVjFEQuOLm5WwCkVnZftiHlIdCrxdax3s,21297
 nshtrainer/model/config.py,sha256=6lATW6-Z1SIDgQ1IWrGBVQKTr8DhL5b_rFbJHQz0d5o,66796
 nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
 nshtrainer/model/modules/debug.py,sha256=DTVty8cKnzj1GCULRyGx_sWTTsq9NLi30dzqjRTnuCU,1127
@@ -72,6 +72,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.8.3.dist-info/METADATA,sha256=9sliOedwTYAjaDReKXVYsZYzLy9qF26V7Qm8EGa06CY,647
-nshtrainer-0.8.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.8.3.dist-info/RECORD,,
+nshtrainer-0.8.5.dist-info/METADATA,sha256=swroihBvsK71kkd51ekyULjRaBGr3ujP9fZ4HDQqhN8,647
+nshtrainer-0.8.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.8.5.dist-info/RECORD,,