0.6.3: arraybridge 0.5.0

docs/source/api/backends.rst

@@ -37,7 +37,7 @@ Array payload boundary
 ----------------------
 
 ArrayBridge owns the memory-type taxonomy and framework conversion semantics.
-``polystore.MemoryType`` is a compatibility re-export of
+``polystore.MemoryType`` is a re-export of
 ``arraybridge.MemoryType`` with the same class identity. Disk, Zarr, and
 streaming sinks normalize supported framework arrays through ArrayBridge when
 a host NumPy payload is required; individual storage backends do not maintain

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "polystore"
-version = "0.6.2"
+version = "0.6.3"
 description = "Framework-agnostic multi-backend storage abstraction for ML and scientific computing"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -37,7 +37,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "arraybridge>=0.3.0",
+    "arraybridge>=0.5.0,<0.6",
     "numpy>=1.26.0",
     "portalocker>=2.8.0",  # Cross-platform file locking
     "metaclass-registry>=0.4.0",

src/polystore/array_payload.py

@@ -1,24 +1,15 @@
 """ArrayBridge-backed normalization at persistent storage boundaries."""
 
-from typing import Any
-
 import numpy as np
-from arraybridge import ArrayPayload, convert_memory, detect_memory_type
-from arraybridge.types import MemoryType
+from arraybridge import ArrayPayload
+from arraybridge.libraries import ArrayLibrary
 
 
-def storage_numpy_array(data: Any) -> np.ndarray:
+def storage_numpy_array(data: object) -> np.ndarray:
     """Return one supported array payload as a host NumPy array."""
 
     if isinstance(data, ArrayPayload):
         data = data.array_payload_data()
     if isinstance(data, np.ndarray):
         return data
-    source_type = detect_memory_type(data)
-    converted = convert_memory(
-        data,
-        source_type=source_type,
-        target_type=MemoryType.NUMPY,
-        gpu_id=0,
-    )
-    return np.asarray(converted)
+    return ArrayLibrary.of(data).to_numpy(data)

src/polystore/constants.py

@@ -4,18 +4,7 @@
 
 from enum import Enum
 
-from arraybridge.types import (
-    CPU_MEMORY_TYPES as CPU_MEMORY_TYPES,
-)
-from arraybridge.types import (
-    GPU_MEMORY_TYPES as GPU_MEMORY_TYPES,
-)
-from arraybridge.types import (
-    SUPPORTED_MEMORY_TYPES as SUPPORTED_MEMORY_TYPES,
-)
-from arraybridge.types import (
-    MemoryType as MemoryType,
-)
+from arraybridge.types import MemoryType as MemoryType
 
 
 class Backend(Enum):