arraybridge 0.3.3__tar.gz → 0.3.4__tar.gz

{arraybridge-0.3.3 → arraybridge-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arraybridge
-Version: 0.3.3
+Version: 0.3.4
 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

arraybridge-0.3.4/src/arraybridge/__init__.py

@@ -0,0 +1,109 @@
+"""
+arraybridge: Unified API for NumPy, CuPy, PyTorch, TensorFlow, JAX, and pyclesperanto.
+
+This package provides automatic memory type conversion, declarative decorators,
+and unified utilities for working with multiple array/tensor frameworks.
+
+Package-root imports stay lightweight: submodule exports load on first
+attribute access instead of at import time. Declaration-only consumers
+(memory-type tables, configuration modules) therefore do not pay the
+NumPy/numcodecs import cost.
+"""
+
+__version__ = "0.3.4"
+
+_LAZY_EXPORTS: dict[str, str] = {
+    "MemoryType": ".types",
+    "MemoryContractAttribute": ".types",
+    "ArrayPayload": ".array_payload",
+    "ArrayGeometry": ".array_geometry",
+    "CPU_MEMORY_TYPES": ".types",
+    "GPU_MEMORY_TYPES": ".types",
+    "SUPPORTED_MEMORY_TYPES": ".types",
+    "convert_memory": ".converters",
+    "detect_memory_type": ".converters",
+    "memory_types": ".decorators",
+    "DtypeConversion": ".decorators",
+    "SliceBySliceRuntimeParameter": ".decorators",
+    "wrap_dtype_preserving_callable": ".decorators",
+    "stack_slices": ".stack_utils",
+    "unstack_slices": ".stack_utils",
+    "process_slices": ".slice_processing",
+    "cleanup_all_gpu_frameworks": ".gpu_cleanup",
+    "MemoryConversionError": ".exceptions",
+    "SCALING_FUNCTIONS": ".dtype_scaling",
+    "_FRAMEWORK_CONFIG": ".framework_config",
+    "_FRAMEWORK_OPS": ".framework_ops",
+    "_execute_with_oom_recovery": ".oom_recovery",
+    "_ensure_module": ".utils",
+    "_supports_dlpack": ".utils",
+    "_get_device_id": ".utils",
+    # Decorator exports named after each memory type (numpy, cupy, torch, ...).
+    "numpy": ".decorators",
+    "cupy": ".decorators",
+    "torch": ".decorators",
+    "tensorflow": ".decorators",
+    "jax": ".decorators",
+    "pyclesperanto": ".decorators",
+}
+
+__all__ = [
+    # Types
+    "MemoryType",
+    "MemoryContractAttribute",
+    "ArrayPayload",
+    "ArrayGeometry",
+    "CPU_MEMORY_TYPES",
+    "GPU_MEMORY_TYPES",
+    "SUPPORTED_MEMORY_TYPES",
+    # Converters
+    "convert_memory",
+    "detect_memory_type",
+    # Decorators
+    "memory_types",
+    "DtypeConversion",
+    "SliceBySliceRuntimeParameter",
+    "wrap_dtype_preserving_callable",
+    # Stack utilities
+    "stack_slices",
+    "unstack_slices",
+    # Slice processing
+    "process_slices",
+    # GPU cleanup
+    "cleanup_all_gpu_frameworks",
+    # Exceptions
+    "MemoryConversionError",
+    # Scaling
+    "SCALING_FUNCTIONS",
+    # Framework config (internal but needed by some consumers)
+    "_FRAMEWORK_CONFIG",
+    "_FRAMEWORK_OPS",
+    # OOM recovery
+    "_execute_with_oom_recovery",
+    # Utils
+    "_ensure_module",
+    "_supports_dlpack",
+    "_get_device_id",
+    "numpy",
+    "cupy",
+    "torch",
+    "tensorflow",
+    "jax",
+    "pyclesperanto",
+]
+
+
+def __getattr__(name: str):
+    module_name = _LAZY_EXPORTS.get(name)
+    if module_name is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+    import importlib
+
+    value = getattr(importlib.import_module(module_name, __name__), name)
+    globals()[name] = value
+    return value
+
+
+def __dir__() -> list[str]:
+    return sorted(set(globals()) | set(_LAZY_EXPORTS))

{arraybridge-0.3.3 → arraybridge-0.3.4}/src/arraybridge/types.py

@@ -5,6 +5,8 @@ This module defines the MemoryType enum and related constants for managing
 different array/tensor frameworks.
 """
 
+from __future__ import annotations
+
 import importlib
 import importlib.util
 import logging
@@ -15,17 +17,10 @@ from contextlib import AbstractContextManager, contextmanager, nullcontext
 from dataclasses import dataclass
 from enum import Enum
 from pathlib import Path
-from typing import Any, TypeVar, cast
-
-from arraybridge.array_operations import (
-    CUPY_OPERATIONS,
-    JAX_OPERATIONS,
-    NUMPY_OPERATIONS,
-    PYCLESPERANTO_OPERATIONS,
-    TENSORFLOW_OPERATIONS,
-    TORCH_OPERATIONS,
-    ArrayOperations,
-)
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+if TYPE_CHECKING:
+    from arraybridge.array_operations import ArrayOperations
 
 T = TypeVar("T")
 logger = logging.getLogger(__name__)
@@ -561,7 +556,7 @@ class _MemoryTypeFields:
     module_aliases: tuple[str, ...]
     import_environment: tuple[tuple[str, str], ...]
     _runtime: FrameworkRuntime
-    _operations: ArrayOperations
+    _operations_name: str
 
 
 class MemoryType(_MemoryTypeFields, Enum):
@@ -571,7 +566,7 @@ class MemoryType(_MemoryTypeFields, Enum):
         cls,
         value: str,
         *declaration: Any,
-    ) -> "MemoryType":
+    ) -> MemoryType:
         (
             import_name,
             display_name,
@@ -589,9 +584,18 @@ class MemoryType(_MemoryTypeFields, Enum):
         member.module_aliases = cast(tuple[str, ...], module_aliases)
         member.import_environment = cast(tuple[tuple[str, str], ...], import_environment)
         member._runtime = cast(FrameworkRuntime, runtime)
-        member._operations = cast(ArrayOperations, operations)
+        member._operations_name = cast(str, operations)
         return member
 
+    @property
+    def _operations(self) -> ArrayOperations:
+        """Resolve this declaration's operations registry on first access."""
+
+        from arraybridge import array_operations
+
+        operations: ArrayOperations = getattr(array_operations, self._operations_name)
+        return operations
+
     NUMPY = (
         "numpy",
         "numpy",
@@ -600,7 +604,7 @@ class MemoryType(_MemoryTypeFields, Enum):
         (),
         (),
         FrameworkRuntime(oom_matcher=_numpy_oom),
-        NUMPY_OPERATIONS,
+        "NUMPY_OPERATIONS",
     )
     CUPY = (
         "cupy",
@@ -624,7 +628,7 @@ class MemoryType(_MemoryTypeFields, Enum):
             oom_matcher=_cupy_oom,
             subprocess_environment=_nvidia_wheel_subprocess_environment,
         ),
-        CUPY_OPERATIONS,
+        "CUPY_OPERATIONS",
     )
     TORCH = (
         "torch",
@@ -647,7 +651,7 @@ class MemoryType(_MemoryTypeFields, Enum):
             dlpack_exporter=_protocol_dlpack_export,
             oom_matcher=_torch_oom,
         ),
-        TORCH_OPERATIONS,
+        "TORCH_OPERATIONS",
     )
     TENSORFLOW = (
         "tensorflow",
@@ -666,7 +670,7 @@ class MemoryType(_MemoryTypeFields, Enum):
             dlpack_validator=_tensorflow_dlpack,
             oom_matcher=_tensorflow_oom,
         ),
-        TENSORFLOW_OPERATIONS,
+        "TENSORFLOW_OPERATIONS",
     )
     JAX = (
         "jax",
@@ -684,7 +688,7 @@ class MemoryType(_MemoryTypeFields, Enum):
             dlpack_exporter=_protocol_dlpack_export,
             oom_matcher=_jax_oom,
         ),
-        JAX_OPERATIONS,
+        "JAX_OPERATIONS",
     )
     PYCLESPERANTO = (
         "pyclesperanto",
@@ -701,7 +705,7 @@ class MemoryType(_MemoryTypeFields, Enum):
             move_to_active_device=_pyclesperanto_device_move,
             oom_matcher=_pyclesperanto_oom,
         ),
-        PYCLESPERANTO_OPERATIONS,
+        "PYCLESPERANTO_OPERATIONS",
     )
 
     @property
@@ -953,7 +957,7 @@ class MemoryType(_MemoryTypeFields, Enum):
         payload = data if isinstance(data, DLPackPayload) else DLPackPayload(data, data)
         return importer(payload, framework)
 
-    def convert_to(self, data: Any, target: "MemoryType", device_id: int) -> Any:
+    def convert_to(self, data: Any, target: MemoryType, device_id: int) -> Any:
         """Convert one array through the source and target declarations."""
 
         if self is target:

arraybridge-0.3.3/src/arraybridge/__init__.py

@@ -1,75 +0,0 @@
-"""
-arraybridge: Unified API for NumPy, CuPy, PyTorch, TensorFlow, JAX, and pyclesperanto.
-
-This package provides automatic memory type conversion, declarative decorators,
-and unified utilities for working with multiple array/tensor frameworks.
-"""
-
-__version__ = "0.3.3"
-
-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
-from .exceptions import MemoryConversionError
-from .framework_config import _FRAMEWORK_CONFIG
-from .framework_ops import _FRAMEWORK_OPS
-from .gpu_cleanup import cleanup_all_gpu_frameworks
-from .oom_recovery import _execute_with_oom_recovery
-from .slice_processing import process_slices
-from .stack_utils import stack_slices, unstack_slices
-from .types import (
-    CPU_MEMORY_TYPES,
-    GPU_MEMORY_TYPES,
-    SUPPORTED_MEMORY_TYPES,
-    MemoryContractAttribute,
-    MemoryType,
-)
-from .utils import _ensure_module, _get_device_id, _supports_dlpack
-
-DtypeConversion = _decorators.DtypeConversion
-SliceBySliceRuntimeParameter = _decorators.SliceBySliceRuntimeParameter
-memory_types = _decorators.memory_types
-wrap_dtype_preserving_callable = _decorators.wrap_dtype_preserving_callable
-for _memory_type in MemoryType:
-    globals()[_memory_type.value] = getattr(_decorators, _memory_type.value)
-
-__all__ = [
-    # Types
-    "MemoryType",
-    "MemoryContractAttribute",
-    "ArrayPayload",
-    "ArrayGeometry",
-    "CPU_MEMORY_TYPES",
-    "GPU_MEMORY_TYPES",
-    "SUPPORTED_MEMORY_TYPES",
-    # Converters
-    "convert_memory",
-    "detect_memory_type",
-    # Decorators
-    "memory_types",
-    "DtypeConversion",
-    "SliceBySliceRuntimeParameter",
-    "wrap_dtype_preserving_callable",
-    # Stack utilities
-    "stack_slices",
-    "unstack_slices",
-    # Slice processing
-    "process_slices",
-    # GPU cleanup
-    "cleanup_all_gpu_frameworks",
-    # Exceptions
-    "MemoryConversionError",
-    # Scaling
-    "SCALING_FUNCTIONS",
-    # Framework config (internal but needed by some consumers)
-    "_FRAMEWORK_CONFIG",
-    "_FRAMEWORK_OPS",
-    # OOM recovery
-    "_execute_with_oom_recovery",
-    # Utils
-    "_ensure_module",
-    "_supports_dlpack",
-    "_get_device_id",
-] + [memory_type.value for memory_type in MemoryType]