nshtrainer 0.14.0__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 (5) hide show
  1. nshtrainer/runner.py +2 -3
  2. nshtrainer/trainer/trainer.py +1 -1
  3. {nshtrainer-0.14.0.dist-info → nshtrainer-0.14.2.dist-info}/METADATA +1 -1
  4. {nshtrainer-0.14.0.dist-info → nshtrainer-0.14.2.dist-info}/RECORD +5 -5
  5. {nshtrainer-0.14.0.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/trainer/trainer.py CHANGED
@@ -244,7 +244,7 @@ class Trainer(LightningTrainer):
244
244
  log.critical(f"Disabling logger because {config.trainer.logging.enabled=}.")
245
245
  kwargs["logger"] = False
246
246
  else:
247
- _update_kwargs(logger=config.trainer.logging.create_loggers(config))
247
+ _update_kwargs(logger=list(config.trainer.logging.create_loggers(config)))
248
248
 
249
249
  if config.trainer.auto_determine_num_nodes:
250
250
  # When num_nodes is auto, we need to detect the number of nodes.
{nshtrainer-0.14.0.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.0
3
+ Version: 0.14.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.14.0.dist-info → nshtrainer-0.14.2.dist-info}/RECORD RENAMED
@@ -73,13 +73,13 @@ 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
80
80
  nshtrainer/trainer/checkpoint_connector.py,sha256=F2tkHogbMAa5U7335sm77sZBkjEDa5v46XbJCH9Mg6c,2167
81
81
  nshtrainer/trainer/signal_connector.py,sha256=llwc8pdKAWxREFpjdi14Bpy8rGVMEJsmJx_s2p4gI8E,10689
82
- nshtrainer/trainer/trainer.py,sha256=IHEtuDVVBradVQOKSP9zYAalkn2sguXUZixzvS8P4UY,17097
82
+ nshtrainer/trainer/trainer.py,sha256=M97phnALfG18VxkMLoDr5AKFf4UaPBdc6S2BghdBtas,17103
83
83
  nshtrainer/util/_environment_info.py,sha256=yPtAbgjCY4tkvh5wp9sjNsF0Z45TYwzEAM_N2_b5BbY,23123
84
84
  nshtrainer/util/_useful_types.py,sha256=dwZokFkIe7M5i2GR3nQ9A1lhGw06DMAFfH5atyquqSA,8000
85
85
  nshtrainer/util/environment.py,sha256=AeW_kLl-N70wmb6L_JLz1wRj0kA70xs6RCmc9iUqczE,4159
@@ -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.0.dist-info/METADATA,sha256=9nlcq0ehVp4Afj3J7xMRMn1b4kOs7oVBCoaJqGYj1DU,860
91
- nshtrainer-0.14.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
92
- nshtrainer-0.14.0.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,,