nshutils 0.20.0__tar.gz → 0.22.0__tar.gz

{nshutils-0.20.0 → nshutils-0.22.0}/PKG-INFO

@@ -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 → nshutils-0.22.0}/pyproject.toml

@@ -1,9 +1,9 @@
 [project]
 name = "nshutils"
-version = "0.20.0"
+version = "0.22.0"
 description = ""
 authors = [{ name = "Nima Shoghi", email = "nimashoghi@gmail.com" }]
-requires-python = ">=3.10,<4.0"
+requires-python = ">=3.9,<4.0"
 readme = "README.md"
 
 dependencies = [
@@ -13,13 +13,12 @@ dependencies = [
     "jaxtyping",
     "beartype",
     "uuid7",
-    "nshconfig",
 ]
 
 [project.optional-dependencies]
 snoop = ["pysnooper"]
-pprint = ["rich[jupyter]", "treescope"]
-extra = ["pysnooper", "rich[jupyter]", "treescope"]
+pprint = ["rich[jupyter]", "treescope; python_version >= '3.10'"]
+extra = ["pysnooper", "rich[jupyter]", "treescope; python_version >= '3.10'"]
 
 [project.urls]
 homepage = "https://github.com/nimashoghi/nshutils"

{nshutils-0.20.0 → nshutils-0.22.0}/src/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-0.20.0 → nshutils-0.22.0}/src/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