nshutils 0.22.0__py3-none-any.whl → 0.22.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. nshutils/actsave/_saver.py +3 -3
  2. nshutils/lovely/_monkey_patch_all.py +11 -12
  3. {nshutils-0.22.0.dist-info → nshutils-0.22.2.dist-info}/METADATA +1 -1
  4. {nshutils-0.22.0.dist-info → nshutils-0.22.2.dist-info}/RECORD +5 -5
  5. {nshutils-0.22.0.dist-info → nshutils-0.22.2.dist-info}/WHEEL +0 -0
nshutils/actsave/_saver.py CHANGED
@@ -9,7 +9,7 @@ from dataclasses import dataclass
9
9
  from functools import wraps
10
10
  from logging import getLogger
11
11
  from pathlib import Path
12
- from typing import TYPE_CHECKING, Generic, Literal, cast, overload
12
+ from typing import TYPE_CHECKING, Generic, Literal, Union, cast, overload
13
13
 
14
14
  import numpy as np
15
15
  from typing_extensions import Never, ParamSpec, TypeAliasType, TypeVar, override
@@ -37,9 +37,9 @@ else:
37
37
  log = getLogger(__name__)
38
38
 
39
39
  Value = TypeAliasType(
40
- "Value", int | float | complex | bool | str | np.ndarray | Tensor | None
40
+ "Value", Union[int, float, complex, bool, str, np.ndarray, Tensor, None]
41
41
  )
42
- ValueOrLambda = TypeAliasType("ValueOrLambda", Value | Callable[..., Value])
42
+ ValueOrLambda = TypeAliasType("ValueOrLambda", Union[Value, Callable[..., Value]])
43
43
 
44
44
 
45
45
  def _torch_is_scripting() -> bool:
nshutils/lovely/_monkey_patch_all.py CHANGED
@@ -41,21 +41,20 @@ def monkey_patch(libraries: list[Library] | Literal["auto"] = "auto"):
41
41
 
42
42
  with contextlib.ExitStack() as stack:
43
43
  for library in libraries:
44
- match library:
45
- case "torch":
46
- from .torch_ import torch_monkey_patch
44
+ if library == "torch":
45
+ from .torch_ import torch_monkey_patch
47
46
 
48
- stack.enter_context(torch_monkey_patch())
49
- case "jax":
50
- from .jax_ import jax_monkey_patch
47
+ stack.enter_context(torch_monkey_patch())
48
+ elif library == "jax":
49
+ from .jax_ import jax_monkey_patch
51
50
 
52
- stack.enter_context(jax_monkey_patch())
53
- case "numpy":
54
- from .numpy_ import numpy_monkey_patch
51
+ stack.enter_context(jax_monkey_patch())
52
+ elif library == "numpy":
53
+ from .numpy_ import numpy_monkey_patch
55
54
 
56
- stack.enter_context(numpy_monkey_patch())
57
- case _:
58
- assert_never(library)
55
+ stack.enter_context(numpy_monkey_patch())
56
+ else:
57
+ assert_never(library)
59
58
 
60
59
  log.info(
61
60
  f"Monkey patched libraries: {', '.join(libraries)}. "
{nshutils-0.22.0.dist-info → nshutils-0.22.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: nshutils
3
- Version: 0.22.0
3
+ Version: 0.22.2
4
4
  Summary:
5
5
  Author: Nima Shoghi
6
6
  Author-email: nimashoghi@gmail.com
{nshutils-0.22.0.dist-info → nshutils-0.22.2.dist-info}/RECORD RENAMED
@@ -2,13 +2,13 @@ nshutils/__init__.py,sha256=AFx1d5k34MyJ2kCHQL5vrZB8GDp2nYUaIUEjszSa25I,477
2
2
  nshutils/__init__.pyi,sha256=ICbY2_XBAlXIVOGyK4PQpatmlUFHHc5-bqM4sfFZoAY,613
3
3
  nshutils/actsave/__init__.py,sha256=hAVsog9d1g3_rQN1TRslrl6sK1PhCGbjy8PPUAmJI58,203
4
4
  nshutils/actsave/_loader.py,sha256=mof3HezUNvLliz7macstX6ewXW05L0Mtv3zJyrbmImg,4640
5
- nshutils/actsave/_saver.py,sha256=LulgC_B7oYtsGjW_I_pnSLrf1k9lmMvHqRxcmfqqrjU,10441
5
+ nshutils/actsave/_saver.py,sha256=Yu0z7D8RPxc-drieVXvxAwFaaNPJKwMQWBj3zwSm8c0,10454
6
6
  nshutils/collections.py,sha256=QWGyANmo4Efq4XRNHDSTE9tRLStwEZHGwE0ATHR-Vqo,5233
7
7
  nshutils/display.py,sha256=Ge63yllx7gi-MKL3mKQeQ5doql_nj56-o5aoTVmusDg,1473
8
8
  nshutils/logging.py,sha256=78pv3-I_gmbKSf5_mYYBr6_H4GNBGErghAdhH9wfYIc,2205
9
9
  nshutils/lovely/__init__.py,sha256=gbWMNs7xfK1CiNdkHvfH0KcyaGjdZ8_WUBGfaEUDN4I,451
10
10
  nshutils/lovely/_base.py,sha256=c2XxNJlEdET2mP2gLzlYY1KHsEN4H9eDD_x8SptuBTA,4277
11
- nshutils/lovely/_monkey_patch_all.py,sha256=WZsC6Xp5-Z2GBd6xyZZEsD4C2xNyZy0YBfjZzAy3m8M,2028
11
+ nshutils/lovely/_monkey_patch_all.py,sha256=zgMupp2Wc_O9R3arl-BAIePpvQSi6TCeshGMaui-Cc8,1986
12
12
  nshutils/lovely/config.py,sha256=lVNMuU1oUvsYlGN0Sn-m6iOLbJIchVnWDpyHm09nWo8,1224
13
13
  nshutils/lovely/jax_.py,sha256=mPH-tSOzWE27ymupllBnO4O6avvJJxpF2a1-G4dN5Ow,2214
14
14
  nshutils/lovely/numpy_.py,sha256=iHA4BCJIW9IU6DXKEfYbh9RA2xyeXeL0tyHdEja89Sw,1866
@@ -17,6 +17,6 @@ nshutils/lovely/utils.py,sha256=2ksT5YGVViFuWc8jSkwVCsABripJmyVJdEDDH7aab70,1045
17
17
  nshutils/snoop.py,sha256=7d7_Q5sJmINL1J29wcnxEvpV95zvZYNoVn5frCq-rww,7393
18
18
  nshutils/typecheck.py,sha256=Gi7xtfilN_UwZ1FTFqBVKDhcQzBEDonVxIv3bUj-uXY,5582
19
19
  nshutils/util.py,sha256=tx-XiRbOrpafV3OkJDE5IVFtzn3kN7uSZ8FkMor0H5c,2845
20
- nshutils-0.22.0.dist-info/METADATA,sha256=_N5Hzlby5_zpzat8Dan7ws-ze4ln0On2AcSekA4Te4c,4406
21
- nshutils-0.22.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
22
- nshutils-0.22.0.dist-info/RECORD,,
20
+ nshutils-0.22.2.dist-info/METADATA,sha256=Fyf30EVxNxg10Vrrg3lEXnTXA5aByv2UBtMqBS6LEg4,4406
21
+ nshutils-0.22.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
22
+ nshutils-0.22.2.dist-info/RECORD,,