nshutils 0.20.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

nshutils/actsave/_saver.py

@@ -9,7 +9,7 @@ from dataclasses import dataclass
 from functools import wraps
 from logging import getLogger
 from pathlib import Path
-from typing import TYPE_CHECKING, Generic, TypeAlias, cast, overload
+from typing import TYPE_CHECKING, Generic, Literal, cast, overload
 
 import numpy as np
 from typing_extensions import Never, ParamSpec, TypeAliasType, TypeVar, override
@@ -22,15 +22,17 @@ if not TYPE_CHECKING:
         import torch  # type: ignore
 
         Tensor = torch.Tensor
+        _torch_installed = True
     except ImportError:
         torch = None
+        _torch_installed = False
 
         Tensor = Never
 else:
     import torch  # type: ignore
 
     Tensor = torch.Tensor
-
+    _torch_installed: Literal[True] = True
 
 log = getLogger(__name__)
 
@@ -41,7 +43,7 @@ ValueOrLambda = TypeAliasType("ValueOrLambda", Value | Callable[..., Value])
 
 
 def _torch_is_scripting() -> bool:
-    if torch is None:
+    if _torch_installed:
         return False
 
     return torch.jit.is_scripting()
@@ -56,7 +58,7 @@ def _to_numpy(activation: Value) -> np.ndarray:
         return np.array(activation)
     elif isinstance(activation, np.ndarray):
         return activation
-    elif isinstance(activation, Tensor):
+    elif _torch_installed and isinstance(activation, Tensor):
         activation_ = activation.detach()
         if activation_.is_floating_point():
             # NOTE: We need to convert to float32 because [b]float16 is not supported by numpy

nshutils/lovely/config.py

@@ -1,15 +1,15 @@
 from __future__ import annotations
 
+from dataclasses import dataclass
 from typing import ClassVar
 
-import nshconfig as C
 from typing_extensions import Self
 
 
-class LovelyConfig(C.Config):
+@dataclass
+class LovelyConfig:
     """
     This class is used to manage the configuration of the Lovely library.
-    It inherits from the Config class in the nshconfig module.
     """
 
     precision: int = 3

{nshutils-0.20.0.dist-info → nshutils-0.22.0.dist-info}/METADATA

@@ -1,11 +1,12 @@
 Metadata-Version: 2.3
 Name: nshutils
-Version: 0.20.0
+Version: 0.22.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.9,<4.0
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -16,14 +17,13 @@ Provides-Extra: snoop
 Requires-Dist: beartype
 Requires-Dist: jaxtyping
 Requires-Dist: lazy-loader
-Requires-Dist: nshconfig
 Requires-Dist: numpy
 Requires-Dist: pysnooper ; extra == "extra"
 Requires-Dist: pysnooper ; extra == "snoop"
 Requires-Dist: rich[jupyter] ; extra == "extra"
 Requires-Dist: rich[jupyter] ; extra == "pprint"
-Requires-Dist: treescope ; extra == "extra"
-Requires-Dist: treescope ; extra == "pprint"
+Requires-Dist: treescope ; (python_version >= "3.10") and (extra == "extra")
+Requires-Dist: treescope ; (python_version >= "3.10") and (extra == "pprint")
 Requires-Dist: typing-extensions
 Requires-Dist: uuid7
 Project-URL: Homepage, https://github.com/nimashoghi/nshutils

{nshutils-0.20.0.dist-info → nshutils-0.22.0.dist-info}/RECORD

@@ -2,14 +2,14 @@ nshutils/__init__.py,sha256=AFx1d5k34MyJ2kCHQL5vrZB8GDp2nYUaIUEjszSa25I,477
 nshutils/__init__.pyi,sha256=ICbY2_XBAlXIVOGyK4PQpatmlUFHHc5-bqM4sfFZoAY,613
 nshutils/actsave/__init__.py,sha256=hAVsog9d1g3_rQN1TRslrl6sK1PhCGbjy8PPUAmJI58,203
 nshutils/actsave/_loader.py,sha256=mof3HezUNvLliz7macstX6ewXW05L0Mtv3zJyrbmImg,4640
-nshutils/actsave/_saver.py,sha256=pvhtkGALxZqjfVVKthTZYcwgx2PGOBmeLCCNwePSm6A,10312
+nshutils/actsave/_saver.py,sha256=LulgC_B7oYtsGjW_I_pnSLrf1k9lmMvHqRxcmfqqrjU,10441
 nshutils/collections.py,sha256=QWGyANmo4Efq4XRNHDSTE9tRLStwEZHGwE0ATHR-Vqo,5233
 nshutils/display.py,sha256=Ge63yllx7gi-MKL3mKQeQ5doql_nj56-o5aoTVmusDg,1473
 nshutils/logging.py,sha256=78pv3-I_gmbKSf5_mYYBr6_H4GNBGErghAdhH9wfYIc,2205
 nshutils/lovely/__init__.py,sha256=gbWMNs7xfK1CiNdkHvfH0KcyaGjdZ8_WUBGfaEUDN4I,451
 nshutils/lovely/_base.py,sha256=c2XxNJlEdET2mP2gLzlYY1KHsEN4H9eDD_x8SptuBTA,4277
 nshutils/lovely/_monkey_patch_all.py,sha256=WZsC6Xp5-Z2GBd6xyZZEsD4C2xNyZy0YBfjZzAy3m8M,2028
-nshutils/lovely/config.py,sha256=jsUK_kfEvthL94qxpHxM-Xobdv67sWZpnH_ag4HLTNo,1274
+nshutils/lovely/config.py,sha256=lVNMuU1oUvsYlGN0Sn-m6iOLbJIchVnWDpyHm09nWo8,1224
 nshutils/lovely/jax_.py,sha256=mPH-tSOzWE27ymupllBnO4O6avvJJxpF2a1-G4dN5Ow,2214
 nshutils/lovely/numpy_.py,sha256=iHA4BCJIW9IU6DXKEfYbh9RA2xyeXeL0tyHdEja89Sw,1866
 nshutils/lovely/torch_.py,sha256=3wnXLa-1xwuQVk1fM50mBqOVDv5wZHjfnBwzOnjcFjg,2638
@@ -17,6 +17,6 @@ nshutils/lovely/utils.py,sha256=2ksT5YGVViFuWc8jSkwVCsABripJmyVJdEDDH7aab70,1045
 nshutils/snoop.py,sha256=7d7_Q5sJmINL1J29wcnxEvpV95zvZYNoVn5frCq-rww,7393
 nshutils/typecheck.py,sha256=Gi7xtfilN_UwZ1FTFqBVKDhcQzBEDonVxIv3bUj-uXY,5582
 nshutils/util.py,sha256=tx-XiRbOrpafV3OkJDE5IVFtzn3kN7uSZ8FkMor0H5c,2845
-nshutils-0.20.0.dist-info/METADATA,sha256=ha4yqytCPH6u0S3PNQqGBuwCQg9fxuoegr2FN-4ThRo,4316
-nshutils-0.20.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-nshutils-0.20.0.dist-info/RECORD,,
+nshutils-0.22.0.dist-info/METADATA,sha256=_N5Hzlby5_zpzat8Dan7ws-ze4ln0On2AcSekA4Te4c,4406
+nshutils-0.22.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+nshutils-0.22.0.dist-info/RECORD,,

{nshutils-0.20.0.dist-info → nshutils-0.22.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any