nshutils 0.22.3__tar.gz → 0.22.5__tar.gz

{nshutils-0.22.3 → nshutils-0.22.5}/PKG-INFO

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

{nshutils-0.22.3 → nshutils-0.22.5}/pyproject.toml

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

{nshutils-0.22.3 → nshutils-0.22.5}/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, Union, cast, overload
+from typing import TYPE_CHECKING, Any, Generic, Literal, Union, cast, overload
 
 import numpy as np
 from typing_extensions import Never, ParamSpec, TypeAliasType, TypeVar, override
@@ -36,14 +36,15 @@ else:
 
 log = getLogger(__name__)
 
+# Updated to include Any for arbitrary types
 Value = TypeAliasType(
-    "Value", Union[int, float, complex, bool, str, np.ndarray, Tensor, None]
+    "Value", Union[int, float, complex, bool, str, np.ndarray, Tensor, Any, None]
 )
 ValueOrLambda = TypeAliasType("ValueOrLambda", Union[Value, Callable[..., Value]])
 
 
 def _torch_is_scripting() -> bool:
-    if _torch_installed:
+    if not _torch_installed:
         return False
 
     return torch.jit.is_scripting()
@@ -65,9 +66,8 @@ def _to_numpy(activation: Value) -> np.ndarray:
             activation_ = activation_.float()
         return activation_.cpu().numpy()
     else:
-        log.warning(f"Unrecognized activation type {type(activation)}")
-
-    return activation
+        # Handle arbitrary objects using numpy object dtype
+        return np.array(activation, dtype=object)
 
 
 T = TypeVar("T", infer_variance=True)