Apply ruff/refurb rule FURB140

FURB140 Use `itertools.starmap` instead of the generator

Author: DimitriPapadopoulos

src/zarr/abc/store.py

@@ -3,6 +3,7 @@
 from abc import ABC, abstractmethod
 from asyncio import gather
 from typing import TYPE_CHECKING, NamedTuple, Protocol, runtime_checkable
+from itertools import starmap
 
 if TYPE_CHECKING:
     from collections.abc import AsyncGenerator, Iterable
@@ -282,7 +283,7 @@ async def _set_many(self, values: Iterable[tuple[str, Buffer]]) -> None:
         """
         Insert multiple (key, value) pairs into storage.
         """
-        await gather(*(self.set(key, value) for key, value in values))
+        await gather(*starmap(self.set, values))
         return
 
     @property

src/zarr/core/array.py

@@ -80,6 +80,7 @@
 from zarr.registry import get_pipeline_class
 from zarr.storage import StoreLike, make_store_path
 from zarr.storage.common import StorePath, ensure_no_existing_node
+from itertools import starmap
 
 if TYPE_CHECKING:
     from collections.abc import Iterable, Iterator, Sequence
@@ -818,7 +819,7 @@ def cdata_shape(self) -> ChunkCoords:
         Tuple[int]
             The shape of the chunk grid for this array.
         """
-        return tuple(ceildiv(s, c) for s, c in zip(self.shape, self.chunks, strict=False))
+        return tuple(starmap(ceildiv, zip(self.shape, self.chunks, strict=False)))
 
     @property
     def nchunks(self) -> int:
@@ -1397,7 +1398,7 @@ def cdata_shape(self) -> ChunkCoords:
         """
         The shape of the chunk grid for this array.
         """
-        return tuple(ceildiv(s, c) for s, c in zip(self.shape, self.chunks, strict=False))
+        return tuple(starmap(ceildiv, zip(self.shape, self.chunks, strict=False)))
 
     @property
     def nchunks(self) -> int:

src/zarr/core/chunk_grids.py

@@ -188,6 +188,6 @@ def all_chunk_coords(self, array_shape: ChunkCoords) -> Iterator[ChunkCoords]:
     def get_nchunks(self, array_shape: ChunkCoords) -> int:
         return reduce(
             operator.mul,
-            (ceildiv(s, c) for s, c in zip(array_shape, self.chunk_shape, strict=True)),
+            itertools.starmap(ceildiv, zip(array_shape, self.chunk_shape, strict=True)),
             1,
         )

src/zarr/core/common.py

@@ -17,6 +17,7 @@
 import numpy as np
 
 from zarr.core.strings import _STRING_DTYPE
+from itertools import starmap
 
 if TYPE_CHECKING:
     from collections.abc import Awaitable, Callable, Iterator
@@ -52,7 +53,7 @@ async def concurrent_map(
     items: Iterable[T], func: Callable[..., Awaitable[V]], limit: int | None = None
 ) -> list[V]:
     if limit is None:
-        return await asyncio.gather(*[func(*item) for item in items])
+        return await asyncio.gather(*list(starmap(func, items)))
 
     else:
         sem = asyncio.Semaphore(limit)