type checker

zarr-developers · Oct 21, 2024 · 14ee6f9 · 14ee6f9
1 parent 00ecae3
commit 14ee6f9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/virtualizarr/tests/test_writers/test_icechunk.py b/virtualizarr/tests/test_writers/test_icechunk.py
@@ -10,14 +10,14 @@
 import numpy.testing as npt
 from xarray import Dataset, open_dataset
 from xarray.core.variable import Variable
-from zarr import Array, Group, group
+from zarr import Array, Group, group  # type: ignore[import-untyped]
 
 from virtualizarr.manifests import ChunkManifest, ManifestArray
 from virtualizarr.writers.icechunk import dataset_to_icechunk
 from virtualizarr.zarr import ZArray
 
 if TYPE_CHECKING:
-    from icechunk import IcechunkStore
+    from icechunk import IcechunkStore  # type: ignore[import-not-found]
 
 
 @pytest.fixture

diff --git a/virtualizarr/writers/icechunk.py b/virtualizarr/writers/icechunk.py
@@ -4,7 +4,7 @@
 from xarray import Dataset
 from xarray.backends.zarr import encode_zarr_attr_value
 from xarray.core.variable import Variable
-from zarr import Group
+from zarr import Group  # type: ignore[import-untyped]
 
 from virtualizarr.manifests import ChunkManifest, ManifestArray
 from virtualizarr.zarr import encode_dtype
@@ -35,7 +35,7 @@ def dataset_to_icechunk(ds: Dataset, store: "IcechunkStore") -> None:
     ds: xr.Dataset
     store: IcechunkStore
     """
-    from icechunk import IcechunkStore
+    from icechunk import IcechunkStore  # type: ignore[import-not-found]
 
     if not isinstance(store, IcechunkStore):
         raise TypeError(f"expected type IcechunkStore, but got type {type(store)}")