@@ -187,10 +187,7 @@ async def decode_partial(
187
187
Iterable[NDBuffer | None]
188
188
"""
189
189
return await concurrent_map (
190
- [
191
- (byte_getter , selection , chunk_spec )
192
- for byte_getter , selection , chunk_spec in batch_info
193
- ],
190
+ list (batch_info ),
194
191
self ._decode_partial_single ,
195
192
config .get ("async.concurrency" ),
196
193
)
@@ -227,10 +224,7 @@ async def encode_partial(
227
224
The chunk spec contains information about the chunk.
228
225
"""
229
226
await concurrent_map (
230
- [
231
- (byte_setter , chunk_array , selection , chunk_spec )
232
- for byte_setter , chunk_array , selection , chunk_spec in batch_info
233
- ],
227
+ list (batch_info ),
234
228
self ._encode_partial_single ,
235
229
config .get ("async.concurrency" ),
236
230
)
@@ -402,7 +396,7 @@ async def batching_helper(
402
396
batch_info : Iterable [tuple [CodecInput | None , ArraySpec ]],
403
397
) -> list [CodecOutput | None ]:
404
398
return await concurrent_map (
405
- [( chunk_array , chunk_spec ) for chunk_array , chunk_spec in batch_info ] ,
399
+ list ( batch_info ) ,
406
400
noop_for_none (func ),
407
401
config .get ("async.concurrency" ),
408
402
)
0 commit comments