nshtrainer 0.5.1__py3-none-any.whl → 0.5.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 +17 -8
  2. {nshtrainer-0.5.1.dist-info → nshtrainer-0.5.2.dist-info}/METADATA +1 -1
  3. {nshtrainer-0.5.1.dist-info → nshtrainer-0.5.2.dist-info}/RECORD +4 -4
  4. {nshtrainer-0.5.1.dist-info → nshtrainer-0.5.2.dist-info}/WHEEL +0 -0
nshtrainer/runner.py CHANGED
@@ -1,9 +1,9 @@
1
1
  import copy
2
2
  import functools
3
- from collections.abc import Mapping, Sequence
3
+ from collections.abc import Callable, Iterable, Mapping, Sequence
4
4
  from typing import Generic
5
5
 
6
- from nshrunner import RunInfo, RunList, TransformFn
6
+ from nshrunner import RunInfo
7
7
  from nshrunner import Runner as _Runner
8
8
  from nshrunner.snapshot import SnapshotArgType
9
9
  from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
@@ -46,11 +46,14 @@ class Runner(
46
46
 
47
47
  def fast_dev_run(
48
48
  self,
49
- runs: RunList[TConfig, Unpack[TArguments]],
49
+ runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
50
50
  n_batches: int = 1,
51
51
  *,
52
52
  env: Mapping[str, str] | None = None,
53
- transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
53
+ transforms: list[
54
+ Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
55
+ ]
56
+ | None = None,
54
57
  ):
55
58
  transforms = transforms or []
56
59
  transforms.append(
@@ -64,11 +67,14 @@ class Runner(
64
67
 
65
68
  def fast_dev_run_generator(
66
69
  self,
67
- runs: RunList[TConfig, Unpack[TArguments]],
70
+ runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
68
71
  n_batches: int = 1,
69
72
  *,
70
73
  env: Mapping[str, str] | None = None,
71
- transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
74
+ transforms: list[
75
+ Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
76
+ ]
77
+ | None = None,
72
78
  ):
73
79
  transforms = transforms or []
74
80
  transforms.append(
@@ -82,13 +88,16 @@ class Runner(
82
88
 
83
89
  def fast_dev_run_session(
84
90
  self,
85
- runs: RunList[TConfig, Unpack[TArguments]],
91
+ runs: Iterable[tuple[TConfig, Unpack[TArguments]]],
86
92
  n_batches: int = 1,
87
93
  *,
88
94
  snapshot: SnapshotArgType,
89
95
  setup_commands: Sequence[str] | None = None,
90
96
  env: Mapping[str, str] | None = None,
91
- transforms: list[TransformFn[TConfig, Unpack[TArguments]]] | None = None,
97
+ transforms: list[
98
+ Callable[[TConfig, Unpack[TArguments]], tuple[TConfig, Unpack[TArguments]]]
99
+ ]
100
+ | None = None,
92
101
  activate_venv: bool = True,
93
102
  session_name: str = "nshrunner",
94
103
  attach: bool = True,
{nshtrainer-0.5.1.dist-info → nshtrainer-0.5.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.5.1
3
+ Version: 0.5.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.5.1.dist-info → nshtrainer-0.5.2.dist-info}/RECORD RENAMED
@@ -47,7 +47,7 @@ nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,
47
47
  nshtrainer/nn/module_list.py,sha256=fb2u5Rqdjff8Pekyr9hkCPkBorQ-fldzzFAjsgWAm30,1719
48
48
  nshtrainer/nn/nonlinearity.py,sha256=owtU4kh4G98psD0axOJWVfBhm-OtJVgFM-TXSHmbNPU,3625
49
49
  nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
50
- nshtrainer/runner.py,sha256=lBTx7eccoHezRie2GNh9iIJNnjPDV3bsH26wgKQQTqk,3554
50
+ nshtrainer/runner.py,sha256=7EumpnBkdNWjSNT9Gm-pkxAJ3W6-iMC-yae-WNeZcLw,3771
51
51
  nshtrainer/scripts/check_env.py,sha256=IMl6dSqsLYppI0XuCsVq8lK4bYqXwY9KHJkzsShz4Kg,806
52
52
  nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
53
53
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
@@ -59,6 +59,6 @@ nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
59
59
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
60
60
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
61
61
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
62
- nshtrainer-0.5.1.dist-info/METADATA,sha256=ajSyujR9GqfArDI-JfMem74Qae_mQTJpR1quQ2MD4pk,812
63
- nshtrainer-0.5.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
64
- nshtrainer-0.5.1.dist-info/RECORD,,
62
+ nshtrainer-0.5.2.dist-info/METADATA,sha256=7fJ39M-XC3Mp8zOg4AShDvqAyCPSmBZxAIDoKQ50hAs,812
63
+ nshtrainer-0.5.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
64
+ nshtrainer-0.5.2.dist-info/RECORD,,