diff --git a/pyproject.toml b/pyproject.toml index 76a3c668d3..9887c824ca 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -168,24 +168,19 @@ disallow_untyped_decorators = true disallow_any_generics = true disallow_incomplete_defs = true +disallow_untyped_calls = true [[tool.mypy.overrides]] module = [ - "zarr.v2._storage.store", - "zarr.v2._storage.v3_storage_transformers", + "zarr.v2.*", "zarr.group", - "zarr.v2.core", - "zarr.v2.hierarchy", - "zarr.v2.indexing", - "zarr.v2.storage", - "zarr.v2.sync", - "zarr.v2.util", "tests.*", ] check_untyped_defs = false [[tool.mypy.overrides]] module = [ + "zarr.v2.*", "zarr.abc.codec", "zarr.codecs.bytes", "zarr.codecs.pipeline", @@ -194,8 +189,6 @@ module = [ "zarr.array_v2", "zarr.array", "zarr.sync", - "zarr.v2.convenience", - "zarr.v2.meta", ] disallow_any_generics = false @@ -207,6 +200,20 @@ module = [ ] disallow_incomplete_defs = false +[[tool.mypy.overrides]] +module = [ + "zarr.v2.*", + "zarr.array_v2", + "zarr.array", + "zarr.common", + "zarr.store.local", + "zarr.codecs.blosc", + "zarr.codecs.gzip", + "zarr.codecs.zstd", +] +disallow_untyped_calls = false + + [tool.pytest.ini_options] doctest_optionflags = [ "NORMALIZE_WHITESPACE", diff --git a/src/zarr/common.py b/src/zarr/common.py index 6940ec3fe3..7d8431f97e 100644 --- a/src/zarr/common.py +++ b/src/zarr/common.py @@ -79,7 +79,7 @@ class ArraySpec: dtype: np.dtype[Any] fill_value: Any - def __init__(self, shape, dtype, fill_value): + def __init__(self, shape: ChunkCoords, dtype: np.dtype[Any], fill_value: Any) -> None: shape_parsed = parse_shapelike(shape) dtype_parsed = parse_dtype(dtype) fill_value_parsed = parse_fill_value(fill_value) diff --git a/src/zarr/indexing.py b/src/zarr/indexing.py index 75bed63384..9f324eb5ea 100644 --- a/src/zarr/indexing.py +++ b/src/zarr/indexing.py @@ -19,7 +19,7 @@ def _err_too_many_indices(selection: SliceSelection, shape: ChunkCoords) -> None ) -def _err_negative_step(): +def _err_negative_step() -> None: raise IndexError("only slices with step >= 1 are supported") @@ -50,7 +50,7 @@ class _ChunkDimProjection(NamedTuple): dim_out_sel: Optional[slice] -def _ceildiv(a, b): +def _ceildiv(a: float, b: float) -> int: return math.ceil(a / b)