Add as_bytes_like method to Buffer

Author: tomwhite

src/zarr/core/buffer/core.py

@@ -251,6 +251,19 @@ def as_numpy_array(self) -> npt.NDArray[Any]:
         """
         ...
 
+    def as_bytes_like(self) -> BytesLike:
+        """Returns the buffer as a bytes-like object.
+
+        Notes
+        -----
+        Might have to copy data, since the implementation uses `.as_numpy_array()`.
+
+        Returns
+        -------
+            A bytes-like object that implements the Python buffer protocol
+        """
+        return memoryview(self.as_numpy_array().view(np.uint8))  # type: ignore[arg-type]
+
     def to_bytes(self) -> bytes:
         """Returns the buffer as `bytes` (host memory).
 

src/zarr/storage/_obstore.py

@@ -7,8 +7,6 @@
 from collections.abc import Iterable
 from typing import TYPE_CHECKING, Any, TypedDict
 
-import numpy as np
-
 from zarr.abc.store import (
     ByteRequest,
     OffsetByteRequest,
@@ -147,15 +145,15 @@ async def set(self, key: str, value: Buffer) -> None:
 
         self._check_writable()
 
-        buf = value.as_numpy_array().view(np.uint8)
+        buf = value.as_bytes_like()
         await obs.put_async(self.store, key, buf)
 
     async def set_if_not_exists(self, key: str, value: Buffer) -> None:
         # docstring inherited
         import obstore as obs
 
         self._check_writable()
-        buf = value.as_numpy_array().view(np.uint8)
+        buf = value.as_bytes_like()
         with contextlib.suppress(obs.exceptions.AlreadyExistsError):
             await obs.put_async(self.store, key, buf, mode="create")