nshtrainer 0.14.1__py3-none-any.whl → 0.14.2__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 (4) hide show
  1. nshtrainer/runner.py +2 -3
  2. {nshtrainer-0.14.1.dist-info → nshtrainer-0.14.2.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.14.1.dist-info → nshtrainer-0.14.2.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.14.1.dist-info → nshtrainer-0.14.2.dist-info}/WHEEL +0 -0
nshtrainer/runner.py CHANGED
@@ -3,10 +3,9 @@ import functools
3
3
  from collections.abc import Callable, Iterable, Mapping, Sequence
4
4
  from typing import Generic
5
5
 
6
- from nshrunner import RunInfo
6
+ from nshrunner import RunInfo, Snapshot
7
7
  from nshrunner import Runner as _Runner
8
8
  from nshrunner._submit import screen
9
- from nshrunner.snapshot import SnapshotArgType
10
9
  from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
11
10
 
12
11
  from .model.config import BaseConfig
@@ -93,7 +92,7 @@ class Runner(
93
92
  options: screen.ScreenJobKwargs = {},
94
93
  n_batches: int = 1,
95
94
  *,
96
- snapshot: SnapshotArgType,
95
+ snapshot: Snapshot,
97
96
  setup_commands: Sequence[str] | None = None,
98
97
  env: Mapping[str, str] | None = None,
99
98
  transforms: list[
{nshtrainer-0.14.1.dist-info → nshtrainer-0.14.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.14.1
3
+ Version: 0.14.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.14.1.dist-info → nshtrainer-0.14.2.dist-info}/RECORD RENAMED
@@ -73,7 +73,7 @@ nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,
73
73
  nshtrainer/nn/module_list.py,sha256=fb2u5Rqdjff8Pekyr9hkCPkBorQ-fldzzFAjsgWAm30,1719
74
74
  nshtrainer/nn/nonlinearity.py,sha256=owtU4kh4G98psD0axOJWVfBhm-OtJVgFM-TXSHmbNPU,3625
75
75
  nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
76
- nshtrainer/runner.py,sha256=6qfE5FBONzD79kVHuWYKEvK0J_Qi5dMBbHQhRMmnIhE,3649
76
+ nshtrainer/runner.py,sha256=USAjrExHkN5oVNVunsoPnLxfQrEHSaa54S3RipOe544,3605
77
77
  nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
78
78
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
79
79
  nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
@@ -87,6 +87,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
87
87
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
88
88
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
89
89
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
90
- nshtrainer-0.14.1.dist-info/METADATA,sha256=V7kssNBEV-JqILCZbRujHMs4ORomy_hWU4vRK0Je2GU,860
91
- nshtrainer-0.14.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
92
- nshtrainer-0.14.1.dist-info/RECORD,,
90
+ nshtrainer-0.14.2.dist-info/METADATA,sha256=iw_4vM9xNkSXj6rS6R81dRZ76Nqc_bGhK6ayyWAQPoA,860
91
+ nshtrainer-0.14.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
92
+ nshtrainer-0.14.2.dist-info/RECORD,,