Apply ruff/flake8-type-checking rule TCH003

TCH003 Move standard library import into a type-checking block

Author: DimitriPapadopoulos

src/zarr/abc/codec.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 from abc import abstractmethod
-from collections.abc import Awaitable, Callable, Iterable
 from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
 from zarr.abc.metadata import Metadata
@@ -10,6 +9,8 @@
 from zarr.core.config import config
 
 if TYPE_CHECKING:
+    from collections.abc import Awaitable, Callable, Iterable
+
     import numpy as np
     from typing_extensions import Self
 

src/zarr/api/asynchronous.py

@@ -2,7 +2,6 @@
 
 import asyncio
 import warnings
-from collections.abc import Iterable
 from typing import TYPE_CHECKING, Any, Literal, Union, cast
 
 import numpy as np
@@ -18,6 +17,8 @@
 )
 
 if TYPE_CHECKING:
+    from collections.abc import Iterable
+
     from zarr.abc.codec import Codec
     from zarr.core.buffer import NDArrayLike
     from zarr.core.chunk_key_encodings import ChunkKeyEncoding

tests/v3/conftest.py

@@ -1,9 +1,7 @@
 from __future__ import annotations
 
 import pathlib
-from collections.abc import Iterator
 from dataclasses import dataclass, field
-from types import ModuleType
 from typing import TYPE_CHECKING
 
 import numpy as np
@@ -15,6 +13,8 @@
 from zarr.store.remote import RemoteStore
 
 if TYPE_CHECKING:
+    from collections.abc import Iterator
+    from types import ModuleType
     from typing import Any, Literal
 
     from zarr.abc.store import Store

tests/v3/test_indexing.py

@@ -22,12 +22,10 @@
 from zarr.store.common import StorePath
 from zarr.store.memory import MemoryStore
 
-if TYPE_CHECKING:
-    from zarr.abc.store import Store
-
 if TYPE_CHECKING:
     from collections.abc import Iterator
 
+    from zarr.abc.store import Store
     from zarr.core.common import ChunkCoords