nshtrainer 0.15.0__py3-none-any.whl → 0.16.0__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 (6) hide show
  1. nshtrainer/ll/snapshot.py +1 -1
  2. nshtrainer/loggers/wandb.py +4 -4
  3. nshtrainer/scripts/find_packages.py +0 -1
  4. {nshtrainer-0.15.0.dist-info → nshtrainer-0.16.0.dist-info}/METADATA +1 -1
  5. {nshtrainer-0.15.0.dist-info → nshtrainer-0.16.0.dist-info}/RECORD +6 -6
  6. {nshtrainer-0.15.0.dist-info → nshtrainer-0.16.0.dist-info}/WHEEL +0 -0
nshtrainer/ll/snapshot.py CHANGED
@@ -1 +1 @@
1
- from nshrunner.snapshot import * # type: ignore # noqa: F403
1
+ from nshsnap import * # pyright: ignore[reportWildcardImportFromLibrary] # noqa: F403
nshtrainer/loggers/wandb.py CHANGED
@@ -1,9 +1,10 @@
1
+ import importlib.metadata
1
2
  import logging
2
3
  from typing import TYPE_CHECKING, Literal
3
4
 
4
5
  import nshconfig as C
5
- import pkg_resources
6
6
  from lightning.pytorch import Callback, LightningModule, Trainer
7
+ from packaging import version
7
8
  from typing_extensions import override
8
9
 
9
10
  from ..callbacks import WandbWatchConfig
@@ -120,9 +121,8 @@ class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
120
121
  import wandb # type: ignore
121
122
 
122
123
  # The minimum version that supports the new backend is 0.17.5
123
- if pkg_resources.parse_version(
124
- wandb.__version__
125
- ) < pkg_resources.parse_version("0.17.5"):
124
+ wandb_version = version.parse(importlib.metadata.version("wandb"))
125
+ if wandb_version < version.parse("0.17.5"):
126
126
  log.warning(
127
127
  "The version of WandB installed does not support the `wandb-core` backend "
128
128
  f"(expected version >= 0.17.5, found version {wandb.__version__}). "
nshtrainer/scripts/find_packages.py CHANGED
@@ -40,7 +40,6 @@ def main():
40
40
 
41
41
  if args.exclude_std:
42
42
  std_libs = set(sys.stdlib_module_names)
43
- std_libs.update({"pkg_resources"})
44
43
  all_imports = all_imports - std_libs
45
44
 
46
45
  for package in sorted(all_imports):
{nshtrainer-0.15.0.dist-info → nshtrainer-0.16.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: nshtrainer
3
- Version: 0.15.0
3
+ Version: 0.16.0
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshtrainer-0.15.0.dist-info → nshtrainer-0.16.0.dist-info}/RECORD RENAMED
@@ -38,7 +38,7 @@ nshtrainer/ll/model.py,sha256=cxFQfFc-2mAYBGwDpP8m5tjQBs7M47cZ6JoPXksPaoI,473
38
38
  nshtrainer/ll/nn.py,sha256=8qiRDFwojIxkB7-LtNWk4mLL2tJbaskHYofDsOIHiNg,42
39
39
  nshtrainer/ll/optimizer.py,sha256=3T-VZtT73jVvwCNJGDjgGEbzs-1LFTzMQH-SB_58mSo,49
40
40
  nshtrainer/ll/runner.py,sha256=B0m5VEhNKIjF1aFmqPkonkQxDoRL2jeHZGsV3zwhSVE,117
41
- nshtrainer/ll/snapshot.py,sha256=UB3AjU6kjD6UkKcJg0y2m9rHQbjZ78NaRuTdk7TdxUI,62
41
+ nshtrainer/ll/snapshot.py,sha256=EdzvbE6IyHcNpIvkF1lWj3pBigfRF9jQXgOlbgBavYs,87
42
42
  nshtrainer/ll/snoop.py,sha256=hG9VCdm8mIZytHLZgUKRSoWqg55rVvUpBAH-OSiwgvI,36
43
43
  nshtrainer/ll/trainer.py,sha256=hkn2xPtrSPQ7LqQhbyAKuMfNyHdhqB9bDPvgRCK8oJM,47
44
44
  nshtrainer/ll/typecheck.py,sha256=ryV1Tzcf7hJ4I19H1oQVkikU9spmRk8jyIKQZ5UF7pQ,62
@@ -47,7 +47,7 @@ nshtrainer/loggers/__init__.py,sha256=C_xk0A3_qKbNdTmzK85AgjRHFD3w-jPRS2ig-iPhfE
47
47
  nshtrainer/loggers/_base.py,sha256=xiZKEK0ALJkcqf4OpVNRY0QbZsamR_WR7x7m_68YHXQ,705
48
48
  nshtrainer/loggers/csv.py,sha256=D_lYyd94bZ8jAgnRo-ARtFgVcInaD9zktxtsUD9RWCI,1052
49
49
  nshtrainer/loggers/tensorboard.py,sha256=wL2amRSdP68zbslZvBeM0ZQBnjF3hIKsz-_lBbdomaM,2216
50
- nshtrainer/loggers/wandb.py,sha256=QteSPfaBWxQ-FuuaF6_rKxw8twJdg_6oHoTQlbqvrXk,5079
50
+ nshtrainer/loggers/wandb.py,sha256=FPwbf618AYmuPzHdhd1ZFhJ8qDjwTUiSe7cm7g3KCyM,5112
51
51
  nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
52
52
  nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
53
53
  nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=mn6cyizyI_stkXtg6zxIEGF9btIxMRWigUHUTlUYCSw,5221
@@ -71,7 +71,7 @@ nshtrainer/nn/module_list.py,sha256=fb2u5Rqdjff8Pekyr9hkCPkBorQ-fldzzFAjsgWAm30,
71
71
  nshtrainer/nn/nonlinearity.py,sha256=owtU4kh4G98psD0axOJWVfBhm-OtJVgFM-TXSHmbNPU,3625
72
72
  nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
73
73
  nshtrainer/runner.py,sha256=USAjrExHkN5oVNVunsoPnLxfQrEHSaa54S3RipOe544,3605
74
- nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
74
+ nshtrainer/scripts/find_packages.py,sha256=ixYivZobumyyGsf2B9oYMLyLTRcBzY_vUv-u3bNW-hs,1424
75
75
  nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
76
76
  nshtrainer/trainer/_runtime_callback.py,sha256=sd2cUdRJG-UCdQr9ruZvEYpNGNF1t2W2fuxwwVlQD9E,4164
77
77
  nshtrainer/trainer/checkpoint_connector.py,sha256=F2tkHogbMAa5U7335sm77sZBkjEDa5v46XbJCH9Mg6c,2167
@@ -84,6 +84,6 @@ nshtrainer/util/seed.py,sha256=Or2wMPsnQxfnZ2xfBiyMcHFIUt3tGTNeMMyOEanCkqs,280
84
84
  nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
85
85
  nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
86
86
  nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
87
- nshtrainer-0.15.0.dist-info/METADATA,sha256=GWle2LcHPRsK2Ygf8jgf1eAUPa7eaAdiZT_NkVvauNw,885
88
- nshtrainer-0.15.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
89
- nshtrainer-0.15.0.dist-info/RECORD,,
87
+ nshtrainer-0.16.0.dist-info/METADATA,sha256=O8wSotKFVUpCPLCLibdBmC6kUICBBcVMXsThhwuMmKI,885
88
+ nshtrainer-0.16.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
89
+ nshtrainer-0.16.0.dist-info/RECORD,,