Skip to content

Commit 5cc3508

Browse files
committed
Satisfy mypy
1 parent cdd956f commit 5cc3508

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

src/zarr/storage/_object.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -157,22 +157,22 @@ async def get(
157157
try:
158158
if byte_range is None:
159159
resp = await obs.get_async(self.store, key)
160-
return prototype.buffer.from_bytes(await resp.bytes_async())
160+
return prototype.buffer.from_bytes(await resp.bytes_async()) # type: ignore[arg-type]
161161
elif isinstance(byte_range, RangeByteRequest):
162-
resp = await obs.get_range_async(
162+
bytes = await obs.get_range_async(
163163
self.store, key, start=byte_range.start, end=byte_range.end
164164
)
165-
return prototype.buffer.from_bytes(memoryview(resp))
165+
return prototype.buffer.from_bytes(bytes) # type: ignore[arg-type]
166166
elif isinstance(byte_range, OffsetByteRequest):
167167
resp = await obs.get_async(
168168
self.store, key, options={"range": {"offset": byte_range.offset}}
169169
)
170-
return prototype.buffer.from_bytes(await resp.bytes_async())
170+
return prototype.buffer.from_bytes(await resp.bytes_async()) # type: ignore[arg-type]
171171
elif isinstance(byte_range, SuffixByteRequest):
172172
resp = await obs.get_async(
173173
self.store, key, options={"range": {"suffix": byte_range.suffix}}
174174
)
175-
return prototype.buffer.from_bytes(await resp.bytes_async())
175+
return prototype.buffer.from_bytes(await resp.bytes_async()) # type: ignore[arg-type]
176176
else:
177177
raise ValueError(f"Unexpected byte_range, got {byte_range}")
178178
except ALLOWED_EXCEPTIONS:
@@ -260,7 +260,7 @@ def list_dir(self, prefix: str) -> AsyncGenerator[str, None]:
260260

261261

262262
async def _transform_list(
263-
list_stream: AsyncGenerator[list[ObjectMeta], None],
263+
list_stream: ListStream[list[ObjectMeta]],
264264
) -> AsyncGenerator[str, None]:
265265
"""
266266
Transform the result of list into an async generator of paths.
@@ -271,7 +271,7 @@ async def _transform_list(
271271

272272

273273
async def _transform_list_dir(
274-
list_result_coroutine: Coroutine[Any, Any, ListResult], prefix: str
274+
list_result_coroutine: Coroutine[Any, Any, ListResult[list[ObjectMeta]]], prefix: str
275275
) -> AsyncGenerator[str, None]:
276276
"""
277277
Transform the result of list_with_delimiter into an async generator of paths.
@@ -317,7 +317,7 @@ class _OtherRequest(TypedDict):
317317
path: str
318318
"""The path to request from."""
319319

320-
range: OffsetRange | SuffixRange
320+
range: OffsetRange | SuffixRange | None
321321
"""The range request type."""
322322

323323

@@ -353,7 +353,7 @@ async def _make_bounded_requests(
353353
buffer_responses.append(
354354
{
355355
"original_request_index": request["original_request_index"],
356-
"buffer": prototype.buffer.from_bytes(memoryview(response)),
356+
"buffer": prototype.buffer.from_bytes(response), # type: ignore[arg-type]
357357
}
358358
)
359359

@@ -378,7 +378,7 @@ async def _make_other_request(
378378
return [
379379
{
380380
"original_request_index": request["original_request_index"],
381-
"buffer": prototype.buffer.from_bytes(buffer),
381+
"buffer": prototype.buffer.from_bytes(buffer), # type: ignore[arg-type]
382382
}
383383
]
384384

0 commit comments

Comments
 (0)