Fix typing and ignore missing imports

.pre-commit-config.yaml

@@ -30,3 +30,8 @@ repos:
     rev: v4.0.0-alpha.8
     hooks:
       - id: prettier
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.8.0
+    hooks:
+      - id: mypy

ndpyramid/core.py

@@ -7,7 +7,7 @@
 import xarray as xr
 
 from .common import Projection
-from .utils import add_metadata_and_zarr_encoding, get_version, multiscales_template
+from .utils import add_metadata_and_zarr_encoding, get_levels, get_version, multiscales_template
 
 
 def pyramid_coarsen(
@@ -48,7 +48,7 @@ def pyramid_coarsen(
     for key, factor in enumerate(factors):
         # merge dictionary via union operator
         kwargs |= {d: factor for d in dims}
-        plevels[str(key)] = ds.coarsen(**kwargs).mean()
+        plevels[str(key)] = ds.coarsen(**kwargs).mean() # type: ignore
 
     plevels['/'] = xr.Dataset(attrs=attrs)
     return dt.DataTree.from_dict(plevels)
@@ -96,6 +96,9 @@ def pyramid_reproject(
     import rioxarray  # noqa: F401
     from rasterio.warp import Resampling
 
+    if not levels:
+        levels = get_levels(ds)
+
     # multiscales spec
     save_kwargs = {'levels': levels, 'pixels_per_tile': pixels_per_tile}
     attrs = {
@@ -108,9 +111,10 @@ def pyramid_reproject(
         )
     }
 
+
     # Convert resampling from string to dictionary if necessary
     if isinstance(resampling, str):
-        resampling_dict = defaultdict(lambda: resampling)
+        resampling_dict:dict = defaultdict(lambda: resampling)
     else:
         resampling_dict = resampling
 

ndpyramid/utils.py

@@ -30,6 +30,8 @@
 def get_version() -> str:
     return __version__
 
+def get_levels(ds: xr.Dataset) -> int:
+    raise NotImplementedError('Automatic determination of number of levels is not yet implemented')
 
 def multiscales_template(
     *,

pyproject.toml

@@ -109,3 +109,7 @@ known-first-party = ["ndpyramid"]
 [tool.pytest.ini_options]
 console_output_style = "count"
 addopts = "--cov=./ --cov-report=xml --verbose"
+
+[tool.mypy]
+ignore_missing_imports = true
+no_implicit_optional = false