Coverage for src/arraybridge/__init__.py: 100%

1""" 

2arraybridge: Unified API for NumPy, CuPy, PyTorch, TensorFlow, JAX, and pyclesperanto. 

3 

4This package provides automatic memory type conversion, declarative decorators, 

5and unified utilities for working with multiple array/tensor frameworks. 

6""" 

7 

8__version__ = "0.2.0" 

9 

10from .converters import convert_memory, detect_memory_type 

11from .decorators import cupy, jax, memory_types, numpy, tensorflow, torch 

12from .exceptions import MemoryConversionError 

13from .stack_utils import stack_slices, unstack_slices 

14from .types import CPU_MEMORY_TYPES, GPU_MEMORY_TYPES, SUPPORTED_MEMORY_TYPES, MemoryType 

15 

16__all__ = [ 

17 # Types 

18 "MemoryType", 

19 "CPU_MEMORY_TYPES", 

20 "GPU_MEMORY_TYPES", 

21 "SUPPORTED_MEMORY_TYPES", 

22 # Converters 

23 "convert_memory", 

24 "detect_memory_type", 

25 # Decorators 

26 "memory_types", 

27 "numpy", 

28 "cupy", 

29 "torch", 

30 "tensorflow", 

31 "jax", 

32 # Stack utilities 

33 "stack_slices", 

34 "unstack_slices", 

35 # Exceptions 

36 "MemoryConversionError", 

37]