nshutils 0.6.1__tar.gz → 0.7.0__tar.gz

{nshutils-0.6.1 → nshutils-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshutils
-Version: 0.6.1
+Version: 0.7.0
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
@@ -9,10 +9,10 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: beartype (>=0.18.5,<0.19.0)
-Requires-Dist: jaxtyping (>=0.2.33,<0.3.0)
+Requires-Dist: beartype
+Requires-Dist: jaxtyping
 Requires-Dist: numpy
-Requires-Dist: pysnooper (>=1.2.0,<2.0.0)
+Requires-Dist: pysnooper
 Requires-Dist: typing-extensions
 Description-Content-Type: text/markdown
 

{nshutils-0.6.1 → nshutils-0.7.0}/pyproject.toml

@@ -1,17 +1,17 @@
 [tool.poetry]
 name = "nshutils"
-version = "0.6.1"
+version = "0.7.0"
 description = ""
 authors = ["Nima Shoghi <nimashoghi@gmail.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.10"
-pysnooper = "^1.2.0"
-jaxtyping = "^0.2.33"
-typing-extensions = "*"
-beartype = "^0.18.5"
 numpy = "*"
+pysnooper = "*"
+jaxtyping = "*"
+typing-extensions = "*"
+beartype = "*"
 
 
 [tool.poetry.group.dev.dependencies]

{nshutils-0.6.1 → nshutils-0.7.0}/src/nshutils/actsave/_saver.py

@@ -15,24 +15,25 @@ from typing_extensions import Never, ParamSpec, TypeVar, override
 
 from ..collections import apply_to_collection
 
-try:
-    import torch  # type: ignore
+if not TYPE_CHECKING:
+    try:
+        import torch  # type: ignore
 
-    if not TYPE_CHECKING:
         Tensor: TypeAlias = torch.Tensor
-except ImportError:
-    torch = None
+    except ImportError:
+        torch = None
 
-    if not TYPE_CHECKING:
         Tensor: TypeAlias = Never
+else:
+    import torch  # type: ignore
+
+    Tensor: TypeAlias = torch.Tensor
 
-if TYPE_CHECKING:
-    Tensor: TypeAlias = Never
 
 log = getLogger(__name__)
 
 Value: TypeAlias = int | float | complex | bool | str | np.ndarray | Tensor | None
-ValueOrLambda = Value | Callable[..., Value]
+ValueOrLambda: TypeAlias = Value | Callable[..., Value]
 
 
 def _torch_is_scripting() -> bool: