@@ -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
262262async 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
273273async 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