nshtrainer 0.10.0__py3-none-any.whl → 0.10.1__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.
Files changed (8) hide show
  1. nshtrainer/{trainer/_checkpoint_resolver.py → _checkpoint/loader.py} +1 -1
  2. nshtrainer/{trainer/_checkpoint_metadata.py → _checkpoint/metadata.py} +1 -1
  3. nshtrainer/model/config.py +1 -1
  4. nshtrainer/trainer/checkpoint_connector.py +1 -1
  5. nshtrainer/trainer/trainer.py +1 -1
  6. {nshtrainer-0.10.0.dist-info → nshtrainer-0.10.1.dist-info}/METADATA +1 -1
  7. {nshtrainer-0.10.0.dist-info → nshtrainer-0.10.1.dist-info}/RECORD +8 -8
  8. {nshtrainer-0.10.0.dist-info → nshtrainer-0.10.1.dist-info}/WHEEL +0 -0
nshtrainer/{trainer/_checkpoint_resolver.py → _checkpoint/loader.py} RENAMED
@@ -10,7 +10,7 @@ from lightning.pytorch.trainer.states import TrainerFn
10
10
  from typing_extensions import assert_never
11
11
 
12
12
  from ..metrics._config import MetricConfig
13
- from ._checkpoint_metadata import METADATA_PATH_SUFFIX, CheckpointMetadata
13
+ from .metadata import METADATA_PATH_SUFFIX, CheckpointMetadata
14
14
 
15
15
  if TYPE_CHECKING:
16
16
  from ..model.config import BaseConfig
nshtrainer/{trainer/_checkpoint_metadata.py → _checkpoint/metadata.py} RENAMED
@@ -12,7 +12,7 @@ from ..model._environment import EnvironmentConfig
12
12
 
13
13
  if TYPE_CHECKING:
14
14
  from ..model import BaseConfig, LightningModuleBase
15
- from .trainer import Trainer
15
+ from ..trainer.trainer import Trainer
16
16
 
17
17
  log = logging.getLogger(__name__)
18
18
 
nshtrainer/model/config.py CHANGED
@@ -34,6 +34,7 @@ from lightning.pytorch.strategies.strategy import Strategy
34
34
  from pydantic import DirectoryPath
35
35
  from typing_extensions import Self, TypedDict, TypeVar, override
36
36
 
37
+ from .._checkpoint.loader import CheckpointLoadingConfig
37
38
  from ..callbacks import (
38
39
  CallbackConfig,
39
40
  LatestEpochCheckpointCallbackConfig,
@@ -43,7 +44,6 @@ from ..callbacks import (
43
44
  )
44
45
  from ..callbacks.base import CallbackConfigBase
45
46
  from ..metrics import MetricConfig
46
- from ..trainer._checkpoint_resolver import CheckpointLoadingConfig
47
47
  from ._environment import EnvironmentConfig
48
48
 
49
49
  log = getLogger(__name__)
nshtrainer/trainer/checkpoint_connector.py CHANGED
@@ -8,7 +8,7 @@ from lightning.pytorch.trainer.connectors.checkpoint_connector import (
8
8
  from lightning.pytorch.trainer.states import TrainerFn
9
9
  from typing_extensions import override
10
10
 
11
- from ._checkpoint_resolver import CheckpointLoadingConfig, _resolve_checkpoint
11
+ from .._checkpoint.loader import CheckpointLoadingConfig, _resolve_checkpoint
12
12
 
13
13
  if TYPE_CHECKING:
14
14
  from ..model.config import BaseConfig
nshtrainer/trainer/trainer.py CHANGED
@@ -17,6 +17,7 @@ from lightning.pytorch.trainer.states import TrainerFn
17
17
  from lightning.pytorch.utilities.types import _EVALUATE_OUTPUT, _PREDICT_OUTPUT
18
18
  from typing_extensions import Unpack, assert_never, override
19
19
 
20
+ from .._checkpoint.metadata import _write_checkpoint_metadata
20
21
  from ..callbacks.base import resolve_all_callbacks
21
22
  from ..model.config import (
22
23
  AcceleratorConfigProtocol,
@@ -25,7 +26,6 @@ from ..model.config import (
25
26
  LightningTrainerKwargs,
26
27
  StrategyConfigProtocol,
27
28
  )
28
- from ._checkpoint_metadata import _write_checkpoint_metadata
29
29
  from ._runtime_callback import RuntimeTrackerCallback, Stage
30
30
  from .signal_connector import _SignalConnector
31
31
 
{nshtrainer-0.10.0.dist-info → nshtrainer-0.10.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.10.0
3
+ Version: 0.10.1
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.10.0.dist-info → nshtrainer-0.10.1.dist-info}/RECORD RENAMED
@@ -1,4 +1,6 @@
1
1
  nshtrainer/__init__.py,sha256=39loiLLXbaGiozEsAn8mPHopxaPsek8JsgR9DD2gxtY,583
2
+ nshtrainer/_checkpoint/loader.py,sha256=48flPr1XgQHOgIPaCrRqOEvRuG0SZuV3cQ1vgHLqFqI,11025
3
+ nshtrainer/_checkpoint/metadata.py,sha256=C7je_soYyEbZjiq7p2_pSVFkgcXnz2J2H5sMy8oskx0,3051
2
4
  nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
3
5
  nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
4
6
  nshtrainer/_experimental/flops/flop_counter.py,sha256=-sL0Fy6poXa__hyzUMdZScjPULp4coQELQpPU6p6dXU,25736
@@ -50,7 +52,7 @@ nshtrainer/metrics/_config.py,sha256=hWWS4IXENRyH3RmJ7z1Wx1n3Lt1sNMlGOrcU6PW15o0
50
52
  nshtrainer/model/__init__.py,sha256=TbexTxiE20WHYg5q3L88Hysk4LlHeKk_isv33aSBREA,1918
51
53
  nshtrainer/model/_environment.py,sha256=s3JFnigbssFRJTwH33K7DcAYVhLOFCC1OZgFNXJgjuw,22317
52
54
  nshtrainer/model/base.py,sha256=Bmw-t70TydDbE9P0ee-lTibGoUhrCx5Qke-upa7FGVM,17512
53
- nshtrainer/model/config.py,sha256=f8gbTaIi02U8EyooC1vv2ElZfXPgMIAVtU0n-LnkNE4,53187
55
+ nshtrainer/model/config.py,sha256=B1XkKYbhpAm6RmF4n4eR66hMh-kCXwIQB2pQuhR9TZE,53177
54
56
  nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
55
57
  nshtrainer/model/modules/debug.py,sha256=DTVty8cKnzj1GCULRyGx_sWTTsq9NLi30dzqjRTnuCU,1127
56
58
  nshtrainer/model/modules/distributed.py,sha256=ABpR9d-3uBS_fivfy_WYW-dExW6vp5BPaoPQnOudHng,1725
@@ -68,17 +70,15 @@ nshtrainer/runner.py,sha256=6qfE5FBONzD79kVHuWYKEvK0J_Qi5dMBbHQhRMmnIhE,3649
68
70
  nshtrainer/scripts/check_env.py,sha256=IMl6dSqsLYppI0XuCsVq8lK4bYqXwY9KHJkzsShz4Kg,806
69
71
  nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
70
72
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
71
- nshtrainer/trainer/_checkpoint_metadata.py,sha256=dj3g0rUZLWfohIRFAhhLqB4qh1fJsquQ5-EZ0Zbl5ZE,3042
72
- nshtrainer/trainer/_checkpoint_resolver.py,sha256=kfIccBLWAMwn-Bw1pbj3XTXXaCdO_taUEUp3RdwFuLY,11037
73
73
  nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
74
- nshtrainer/trainer/checkpoint_connector.py,sha256=9DrZliK95BIZfwVFxL06Uf7DbfbQ5UAWd0xckH-LU6U,2125
74
+ nshtrainer/trainer/checkpoint_connector.py,sha256=xoqI2dcPnlNFPPLVIU6dBOvRPC9PtfX5qu__xV1lx0Y,2124
75
75
  nshtrainer/trainer/signal_connector.py,sha256=llwc8pdKAWxREFpjdi14Bpy8rGVMEJsmJx_s2p4gI8E,10689
76
- nshtrainer/trainer/trainer.py,sha256=qHwerfdQUCU21IkWf50d_qZAIHb2d8qOLfqTszBpzks,16784
76
+ nshtrainer/trainer/trainer.py,sha256=n3T9Iz3eaDostxEdjapWImAsVMxyU9WBdhlPl0THX-g,16785
77
77
  nshtrainer/util/environment.py,sha256=_SEtiQ_s5bL5pllUlf96AOUv15kNvCPvocVC13S7mIk,4166
78
78
  nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
79
79
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
80
80
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
81
81
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
82
- nshtrainer-0.10.0.dist-info/METADATA,sha256=GslAMAaEXDbMxDd4ijoqjQKYBjb0iAnEGkZ3pAF_sOQ,695
83
- nshtrainer-0.10.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
- nshtrainer-0.10.0.dist-info/RECORD,,
82
+ nshtrainer-0.10.1.dist-info/METADATA,sha256=O8wMPb0ksoZajyes8dsq4IIjsfP_jQaxGYpW3rYE9Ro,695
83
+ nshtrainer-0.10.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
84
+ nshtrainer-0.10.1.dist-info/RECORD,,