Skip to content

Commit cc14e07

Browse files
committed
Update TypeError on GpuMemoryStore
1 parent 1e96600 commit cc14e07

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/zarr/storage/_memory.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -233,8 +233,9 @@ async def set(self, key: str, value: Buffer, byte_range: tuple[int, int] | None
233233
self._check_writable()
234234
assert isinstance(key, str)
235235
if not isinstance(value, Buffer):
236-
raise TypeError(f"Expected Buffer. Got {type(value)}.")
237-
236+
raise TypeError(
237+
f"GpuMemoryStore.set(): `value` must be a Buffer instance. Got an instance of {type(value)} instead."
238+
)
238239
# Convert to gpu.Buffer
239240
gpu_value = value if isinstance(value, gpu.Buffer) else gpu.Buffer.from_buffer(value)
240241
await super().set(key, gpu_value, byte_range=byte_range)

0 commit comments

Comments
 (0)