arraybridge 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

arraybridge/__init__.py

@@ -5,9 +5,10 @@ This package provides automatic memory type conversion, declarative decorators,
 and unified utilities for working with multiple array/tensor frameworks.
 """
 
-__version__ = "0.3.0"
+__version__ = "0.3.2"
 
 from . import decorators as _decorators
+from .array_geometry import ArrayGeometry
 from .array_payload import ArrayPayload
 from .converters import convert_memory, detect_memory_type
 from .dtype_scaling import SCALING_FUNCTIONS
@@ -39,6 +40,7 @@ __all__ = [
     "MemoryType",
     "MemoryContractAttribute",
     "ArrayPayload",
+    "ArrayGeometry",
     "CPU_MEMORY_TYPES",
     "GPU_MEMORY_TYPES",
     "SUPPORTED_MEMORY_TYPES",

arraybridge/array_geometry.py

@@ -0,0 +1,57 @@
+"""Framework-neutral array geometry inspection."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any
+
+import numpy as np
+
+from arraybridge.array_payload import ArrayPayload
+
+
+@dataclass(frozen=True, slots=True)
+class ArrayGeometry:
+    """Concrete shape metadata without moving array data between frameworks."""
+
+    shape: tuple[int, ...]
+
+    @property
+    def ndim(self) -> int:
+        """Return the rank derived from the canonical shape."""
+
+        return len(self.shape)
+
+    @classmethod
+    def from_value(cls, value: Any) -> ArrayGeometry | None:
+        """Inspect an array or nominal payload without forcing host conversion."""
+
+        data = value.array_payload_data() if isinstance(value, ArrayPayload) else value
+        declared_shape = getattr(data, "shape", None)
+        if declared_shape is not None:
+            try:
+                return cls(tuple(int(axis_size) for axis_size in declared_shape))
+            except (TypeError, ValueError):
+                return None
+
+        try:
+            array = np.asarray(data)
+        except (TypeError, ValueError):
+            return None
+        return cls(tuple(int(axis_size) for axis_size in array.shape))
+
+    @classmethod
+    def require_from_value(
+        cls,
+        value: Any,
+        *,
+        value_name: str = "Value",
+    ) -> ArrayGeometry:
+        """Return concrete geometry or reject a value without an array shape."""
+
+        geometry = cls.from_value(value)
+        if geometry is None:
+            raise TypeError(
+                f"{value_name} requires concrete array geometry, got " f"{type(value).__name__}."
+            )
+        return geometry

arraybridge/decorators.py

@@ -23,6 +23,7 @@ from typing import Any, ClassVar, TypeVar, cast
 
 import numpy as np
 from metaclass_registry import AutoRegisterMeta, RegistryFamily, RegistryKeyAttribute
+from python_introspect import RuntimeParameterDeclarationABC
 
 from arraybridge.array_payload import ArrayPayload
 from arraybridge.oom_recovery import _execute_with_oom_recovery
@@ -60,7 +61,7 @@ class DtypeConversion(Enum):
         return dtype_map.get(self, None)
 
 
-class DtypeConversionConfig(ABC):
+class DtypeConversionConfig(RuntimeParameterDeclarationABC):
     """Nominal dtype conversion config surface consumed by decorators."""
 
     @property
@@ -90,7 +91,7 @@ class DtypeConversionConfig(ABC):
         )
 
 
-class SliceBySliceRuntimeParameter:
+class SliceBySliceRuntimeParameter(RuntimeParameterDeclarationABC):
     """Nominal slice-by-slice execution parameter consumed by decorators."""
 
     preserve_for_execution = True

{arraybridge-0.3.0.dist-info → arraybridge-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arraybridge
-Version: 0.3.0
+Version: 0.3.2
 Summary: Unified API for NumPy, CuPy, PyTorch, TensorFlow, JAX, and pyclesperanto with automatic memory type conversion
 Project-URL: Homepage, https://github.com/OpenHCSDev/arraybridge
 Project-URL: Documentation, https://arraybridge.readthedocs.io
@@ -23,6 +23,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Requires-Dist: metaclass-registry>=0.1.5
 Requires-Dist: numpy>=1.20
+Requires-Dist: python-introspect>=0.1.11
 Provides-Extra: all
 Requires-Dist: cupy>=10.0; extra == 'all'
 Requires-Dist: jax>=0.3; extra == 'all'

{arraybridge-0.3.0.dist-info → arraybridge-0.3.2.dist-info}/RECORD

@@ -1,9 +1,10 @@
-arraybridge/__init__.py,sha256=k9fvmnJ--35SWPiNmlFqEJEukWIcEATVom_X3l171Uw,2263
+arraybridge/__init__.py,sha256=xFy8ESRzsIy8ydjdPQs33C5Ed6HH6YUsn65HSmQTtzM,2326
+arraybridge/array_geometry.py,sha256=CopfGmFVKwV5l-lOTImhm1YVh18weUng0ReEF58qlPI,1689
 arraybridge/array_operations.py,sha256=fp69o9BYcCzfhmLcWrH68SVGSfPL3nb6Cty-Amd_oQM,13082
 arraybridge/array_payload.py,sha256=6_jJwPRya0MHrUNSA9kXPlTISpLa0G74rqDValGXKx4,1079
 arraybridge/converters.py,sha256=q44E6Pg1rJz8yq6WQIPLOZPv1lqqcUcEG4Gh4D5M1_o,2649
 arraybridge/converters_registry.py,sha256=_hHtbd-3SMIKUw8fvNu3twPWJY7qHyNyQ_NhVp3eSHQ,5647
-arraybridge/decorators.py,sha256=bJOShfRHZ_HiK4I_ODBALKeJEpnq4PI9kQqwa_uSDFY,19855
+arraybridge/decorators.py,sha256=mYIxBbCuWQVN8dhAE2M7i2Yo0KbcMyY4MZmsfGXGm-0,19975
 arraybridge/dtype_scaling.py,sha256=KDgYyCvxgsQXLJJl_bs5nH46zQimsA7c1Olv6xJ2z1o,831
 arraybridge/exceptions.py,sha256=fhej5ZS39QcBFWzNeNYjoqeLVnOgwwp-cWIme25jyuo,752
 arraybridge/framework_config.py,sha256=5yTTt9NWcwzKDMXbc-fqCdiFl-piKOt3SUnxmRXXCQk,744
@@ -14,7 +15,7 @@ arraybridge/slice_processing.py,sha256=5R10OmHW8BSbJFwT1URd2Sz3P15o5EhaI2VnVUUIv
 arraybridge/stack_utils.py,sha256=FVikN0XG-Xryw-1p9MaSRUTZet7kLcy0gTEERSjc9r4,6915
 arraybridge/types.py,sha256=1otpHqFJfGc9RY40VCzNCkqpTnR8cfckhY44HnuGudI,33191
 arraybridge/utils.py,sha256=RqThtYScPEOEQG491dA8b_VvUTjgcFhwLbh0erQu5Ns,4833
-arraybridge-0.3.0.dist-info/METADATA,sha256=f3bpTWJ_Catw9eGDojwa5yzAeCWJGYQBrbtxwjbAi2o,5234
-arraybridge-0.3.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-arraybridge-0.3.0.dist-info/licenses/LICENSE,sha256=xagEoeTAj1WT64RmyR3E6HH-eTGdgXN6gqPMUUt7L_Y,1070
-arraybridge-0.3.0.dist-info/RECORD,,
+arraybridge-0.3.2.dist-info/METADATA,sha256=tX2dNiYptc7DpXQUqV8x5BX5RzVSMLOD9wM-4xQmg5o,5275
+arraybridge-0.3.2.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+arraybridge-0.3.2.dist-info/licenses/LICENSE,sha256=xagEoeTAj1WT64RmyR3E6HH-eTGdgXN6gqPMUUt7L_Y,1070
+arraybridge-0.3.2.dist-info/RECORD,,