nshutils 0.22.0__tar.gz → 0.22.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: nshutils
-Version: 0.22.0
+Version: 0.22.2
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com

{nshutils-0.22.0 → nshutils-0.22.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nshutils"
-version = "0.22.0"
+version = "0.22.2"
 description = ""
 authors = [{ name = "Nima Shoghi", email = "nimashoghi@gmail.com" }]
 requires-python = ">=3.9,<4.0"

{nshutils-0.22.0 → nshutils-0.22.2}/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, Literal, cast, overload
+from typing import TYPE_CHECKING, Generic, Literal, Union, cast, overload
 
 import numpy as np
 from typing_extensions import Never, ParamSpec, TypeAliasType, TypeVar, override
@@ -37,9 +37,9 @@ else:
 log = getLogger(__name__)
 
 Value = TypeAliasType(
-    "Value", int | float | complex | bool | str | np.ndarray | Tensor | None
+    "Value", Union[int, float, complex, bool, str, np.ndarray, Tensor, None]
 )
-ValueOrLambda = TypeAliasType("ValueOrLambda", Value | Callable[..., Value])
+ValueOrLambda = TypeAliasType("ValueOrLambda", Union[Value, Callable[..., Value]])
 
 
 def _torch_is_scripting() -> bool:

{nshutils-0.22.0 → nshutils-0.22.2}/src/nshutils/lovely/_monkey_patch_all.py

@@ -41,21 +41,20 @@ def monkey_patch(libraries: list[Library] | Literal["auto"] = "auto"):
 
     with contextlib.ExitStack() as stack:
         for library in libraries:
-            match library:
-                case "torch":
-                    from .torch_ import torch_monkey_patch
+            if library == "torch":
+                from .torch_ import torch_monkey_patch
 
-                    stack.enter_context(torch_monkey_patch())
-                case "jax":
-                    from .jax_ import jax_monkey_patch
+                stack.enter_context(torch_monkey_patch())
+            elif library == "jax":
+                from .jax_ import jax_monkey_patch
 
-                    stack.enter_context(jax_monkey_patch())
-                case "numpy":
-                    from .numpy_ import numpy_monkey_patch
+                stack.enter_context(jax_monkey_patch())
+            elif library == "numpy":
+                from .numpy_ import numpy_monkey_patch
 
-                    stack.enter_context(numpy_monkey_patch())
-                case _:
-                    assert_never(library)
+                stack.enter_context(numpy_monkey_patch())
+            else:
+                assert_never(library)
 
         log.info(
             f"Monkey patched libraries: {', '.join(libraries)}. "