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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
nshtrainer/model/base.py CHANGED
@@ -15,7 +15,7 @@ from lightning.fabric.utilities.types import _MAP_LOCATION_TYPE, _PATH
15
15
  from lightning.pytorch import LightningDataModule, LightningModule, Trainer
16
16
  from lightning.pytorch.callbacks import Callback
17
17
  from lightning.pytorch.utilities.types import STEP_OUTPUT
18
- from typing_extensions import Self, TypeVar, deprecated, override
18
+ from typing_extensions import Self, TypeVar, override
19
19
 
20
20
  from .config import (
21
21
  BaseConfig,
@@ -39,12 +39,6 @@ THparams = TypeVar("THparams", bound=BaseConfig, infer_variance=True)
39
39
 
40
40
 
41
41
  class Base(DebugModuleMixin, Generic[THparams]):
42
- @deprecated("Use `ll.nn.MLP` instead.")
43
- def mlp(self, *args, **kwargs):
44
- from ..nn.mlp import MLP
45
-
46
- return MLP(*args, **kwargs)
47
-
48
42
  @torch.jit.unused
49
43
  @property
50
44
  def config(self) -> THparams:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.8.4
3
+ Version: 0.8.5
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
@@ -46,7 +46,7 @@ nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1T
46
46
  nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=mn6cyizyI_stkXtg6zxIEGF9btIxMRWigUHUTlUYCSw,5221
47
47
  nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=h76oTHYpMxauV_l6lviya5DW-WKArwxxf7ZQizhmbCw,2782
48
48
  nshtrainer/model/__init__.py,sha256=y32Hla-5whpzLL2BtCJpBakSp8o-1nQbpO0j_-xq_Po,1864
49
- nshtrainer/model/base.py,sha256=k2Qk_rH4FsBOib2PISXnG97SBlgeD1w_jw02NE4zEQI,21460
49
+ nshtrainer/model/base.py,sha256=YtqnjiMf0cLVjFEQuOLm5WwCkVnZftiHlIdCrxdax3s,21297
50
50
  nshtrainer/model/config.py,sha256=6lATW6-Z1SIDgQ1IWrGBVQKTr8DhL5b_rFbJHQz0d5o,66796
51
51
  nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
52
52
  nshtrainer/model/modules/debug.py,sha256=DTVty8cKnzj1GCULRyGx_sWTTsq9NLi30dzqjRTnuCU,1127
@@ -72,6 +72,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
72
72
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
73
73
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
74
74
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
75
- nshtrainer-0.8.4.dist-info/METADATA,sha256=PS0Z7icSp22Ok5Bs6SnOc0PEBqg42kjXTnpW5jqQaxw,647
76
- nshtrainer-0.8.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
77
- nshtrainer-0.8.4.dist-info/RECORD,,
75
+ nshtrainer-0.8.5.dist-info/METADATA,sha256=swroihBvsK71kkd51ekyULjRaBGr3ujP9fZ4HDQqhN8,647
76
+ nshtrainer-0.8.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
77
+ nshtrainer-0.8.5.dist-info/RECORD,,