rio-tiler 8.0.1__tar.gz → 8.0.3__tar.gz

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rio-tiler
-Version: 8.0.1
+Version: 8.0.3
 Summary: User friendly Rasterio plugin to read raster datasets.
 Project-URL: Homepage, https://cogeotiff.github.io/rio-tiler/
 Project-URL: Documentation, https://cogeotiff.github.io/rio-tiler/

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/pyproject.toml

@@ -70,6 +70,11 @@ docs = [
 deploy = [
     "hatch",
 ]
+notebook = [
+    "IPython",
+    "jupyter",
+    "matplotlib",
+]
 
 [project.optional-dependencies]
 s3 = [
@@ -155,7 +160,7 @@ exclude = [
 max-complexity = 14
 
 [tool.bumpversion]
-current_version = "8.0.1"
+current_version = "8.0.3"
 parse = """(?x)
     (?P<major>\\d+)\\.
     (?P<minor>\\d+)\\.

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/__init__.py

@@ -1,6 +1,6 @@
 """rio-tiler."""
 
-__version__ = "8.0.1"
+__version__ = "8.0.3"
 
 from . import (  # noqa
     colormap,

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/colormap.py

@@ -331,7 +331,7 @@ class ColorMaps:
                 if isinstance(cmap_data, Sequence):
                     cmap_data = [
                         (tuple(inter), parse_color(v))  # type: ignore
-                        for (inter, v) in cmap_data
+                        for (inter, v) in cmap_data  # type: ignore
                     ]
 
             else:

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/experimental/zarr.py

@@ -54,7 +54,7 @@ def open_dataset(src_path: str, **kwargs: Any) -> xarray.Dataset:
     store = obstore.store.from_url(src_path, **kwargs)
     zarr_store = ObjectStore(store=store, read_only=True)
     ds = xarray.open_dataset(
-        zarr_store,
+        zarr_store,  # type: ignore
         decode_times=True,
         decode_coords="all",
         consolidated=True,
@@ -144,7 +144,7 @@ class ZarrReader(BaseReader):
     @property
     def variables(self) -> List[str]:
         """Return dataset variable names"""
-        return list(self.dataset.data_vars)
+        return list(self.dataset.data_vars)  # type: ignore
 
     def _arrange_dims(self, da: xarray.DataArray) -> xarray.DataArray:
         """Arrange coordinates and time dimensions.

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/io/base.py

@@ -5,7 +5,7 @@ import contextlib
 import re
 import warnings
 from functools import cached_property
-from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union
+from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
 import attr
 import numpy
@@ -189,6 +189,8 @@ class SpatialMixin:
                     densify_pts=21,
                 )
 
+        tile_bounds = cast(BBox, tile_bounds)
+
         # If tile_bounds has non-finite value in the dataset CRS we return True
         if not all(numpy.isfinite(tile_bounds)):
             return True
@@ -1115,7 +1117,7 @@ class MultiBandReader(SpatialMixin, metaclass=abc.ABCMeta):
             bands_metadata[band].colorinterp[0] for _, band in enumerate(bands)
         ]
         meta["nodata_type"] = bands_metadata[bands[0]].nodata_type
-        return Info(**meta)
+        return Info.model_validate(meta)
 
     def statistics(
         self,

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/io/rasterio.py

@@ -2,7 +2,7 @@
 
 import contextlib
 import warnings
-from typing import Any, Callable, Dict, List, Optional, Sequence, Union
+from typing import Any, Callable, Dict, List, Optional, Sequence, Union, cast
 
 import attr
 import numpy
@@ -281,7 +281,7 @@ class Reader(BaseReader):
             )
 
         return self.part(
-            tuple(self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z))),
+            cast(BBox, self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z))),
             dst_crs=self.tms.rasterio_crs,
             bounds_crs=self.tms.rasterio_crs,
             height=tilesize,
@@ -667,7 +667,7 @@ class ImageReader(Reader):
             )
 
         return self.part(
-            tuple(self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z))),
+            cast(BBox, self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z))),
             height=tilesize,
             width=tilesize,
             max_size=None,
@@ -810,7 +810,7 @@ class ImageReader(Reader):
         bbox = featureBounds(shape)
 
         # If Image Origin is top Left (non-geo) we need to invert the bbox
-        bbox = [bbox[0], bbox[3], bbox[2], bbox[1]]
+        bbox = (bbox[0], bbox[3], bbox[2], bbox[1])
 
         img = self.part(
             bbox,

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/io/xarray.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import math
 import os
 import warnings
-from typing import Any, Dict, List, Tuple
+from typing import Any, Dict, List, Tuple, cast
 
 import attr
 import numpy
@@ -52,7 +52,7 @@ except ImportError:  # pragma: nocover
     rioxarray = None  # type: ignore
 
 
-MAX_ARRAY_SIZE = os.environ.get("RIO_TILER_MAX_ARRAY_SIZE", 1_000_000_000)  # 1Gb
+MAX_ARRAY_SIZE = int(os.environ.get("RIO_TILER_MAX_ARRAY_SIZE", 1_000_000_000))  # 1Gb
 
 
 @attr.s
@@ -160,7 +160,7 @@ class XarrayReader(BaseReader):
             if coords_name:
                 return [str(self.input.coords[coords_name[0]].data)]
 
-            return [self.input.name or ""]
+            return [self.input.name or ""]  # type: ignore
 
         return [str(band) for d in self._dims for band in self.input[d].values]
 
@@ -281,13 +281,10 @@ class XarrayReader(BaseReader):
                 f"Tile(x={tile_x}, y={tile_y}, z={tile_z}) is outside bounds"
             )
 
-        tile_bounds = tuple(self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z)))
-        dst_crs = self.tms.rasterio_crs
-
         return self.part(
-            tile_bounds,
-            dst_crs=dst_crs,
-            bounds_crs=dst_crs,
+            cast(BBox, self.tms.xy_bounds(Tile(x=tile_x, y=tile_y, z=tile_z))),
+            dst_crs=self.tms.rasterio_crs,
+            bounds_crs=self.tms.rasterio_crs,
             reproject_method=reproject_method,
             auto_expand=auto_expand,
             nodata=nodata,
@@ -568,13 +565,13 @@ class XarrayReader(BaseReader):
 
         arr = da.to_masked_array()
         if out_dtype:
-            arr = arr.astype(out_dtype)
+            arr = cast(numpy.ma.MaskedArray, arr.astype(out_dtype))
         arr.mask |= arr.data == da.rio.nodata
 
         output_bounds = da.rio._unordered_bounds()
         if output_bounds[1] > output_bounds[3] and da.rio.transform().e > 0:
             yaxis = self.input.dims.index(self.input.rio.y_dim)
-            arr = numpy.flip(arr, axis=yaxis)
+            arr = cast(numpy.ma.MaskedArray, numpy.flip(arr, axis=yaxis))
 
         img = ImageData(
             arr,
@@ -645,7 +642,7 @@ class XarrayReader(BaseReader):
 
         if out_dtype:
             arr = arr.astype(out_dtype)
-        arr.mask |= arr.data == da.rio.nodata
+        arr.mask |= arr.data == da.rio.nodata  # type: ignore [attr-defined]
 
         return PointData(
             arr,

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/models.py

@@ -2,7 +2,7 @@
 
 import itertools
 import warnings
-from typing import Any, Dict, List, Literal, Optional, Sequence, Tuple, Union
+from typing import Any, Dict, List, Literal, Optional, Sequence, Tuple, Union, cast
 
 import attr
 import numpy
@@ -104,7 +104,7 @@ def rescale_image(
 ) -> numpy.ma.MaskedArray:
     """Rescale image data in-place."""
     if len(array.shape) < 3:
-        array = numpy.expand_dims(array, axis=0)
+        array = numpy.expand_dims(array, axis=0)  # type: ignore [assignment]
 
     nbands = array.shape[0]
 
@@ -116,14 +116,14 @@ def rescale_image(
 
     for bdx in range(nbands):
         array.data[bdx] = numpy.where(
-            ~array.mask[bdx],
+            ~array.mask[bdx],  # type: ignore [index]
             linear_rescale(
                 array.data[bdx], in_range=in_range[bdx], out_range=out_range[bdx]
             ),
             0,
         )
 
-    return array.astype(out_dtype)
+    return array.astype(out_dtype)  # type: ignore [return-value]
 
 
 def to_masked(array: numpy.ndarray) -> numpy.ma.MaskedArray:
@@ -131,6 +131,7 @@ def to_masked(array: numpy.ndarray) -> numpy.ma.MaskedArray:
     if not numpy.ma.isarray(array):
         array = numpy.ma.asarray(array)
 
+    array = cast(numpy.ma.MaskedArray, array)
     # when a masked array is totally valid, the mask is set to numpy.ma.nomask
     # https://numpy.org/doc/stable/reference/maskedarray.baseclass.html#numpy.ma.nomask
     # doing `array.mask = False` force the creation of the mask array
@@ -156,14 +157,14 @@ class PointData:
 
     array: numpy.ma.MaskedArray = attr.ib(converter=to_masked)
     band_names: List[str] = attr.ib(kw_only=True)
-    band_descriptions: Optional[List[str]] = attr.ib(kw_only=True)
+    band_descriptions: List[str] = attr.ib(kw_only=True)
     coordinates: Optional[Tuple[float, float]] = attr.ib(default=None, kw_only=True)
     crs: Optional[CRS] = attr.ib(default=None, kw_only=True)
     assets: Optional[List] = attr.ib(default=None, kw_only=True)
     metadata: Optional[Dict] = attr.ib(factory=dict, kw_only=True)
     nodata: Optional[NoData] = attr.ib(default=None, kw_only=True)
-    scales: Optional[List[NumType]] = attr.ib(kw_only=True)
-    offsets: Optional[List[NumType]] = attr.ib(kw_only=True)
+    scales: List[NumType] = attr.ib(kw_only=True)
+    offsets: List[NumType] = attr.ib(kw_only=True)
     pixel_location: Optional[Tuple[NumType, NumType]] = attr.ib(
         default=None, kw_only=True
     )
@@ -310,6 +311,8 @@ def masked_and_3d(array: numpy.ndarray) -> numpy.ma.MaskedArray:
     if len(array.shape) < 3:
         array = numpy.expand_dims(array, axis=0)
 
+    array = cast(numpy.ma.MaskedArray, array)
+
     # when a masked array is totally valid, the mask is set to numpy.ma.nomask
     # https://numpy.org/doc/stable/reference/maskedarray.baseclass.html#numpy.ma.nomask
     # doing `array.mask = False` force the creation of the mask array
@@ -344,10 +347,10 @@ class ImageData:
     crs: Optional[CRS] = attr.ib(default=None, kw_only=True)
     metadata: Optional[Dict] = attr.ib(factory=dict, kw_only=True)
     nodata: Optional[NoData] = attr.ib(default=None, kw_only=True)
-    scales: Optional[List[NumType]] = attr.ib(kw_only=True)
-    offsets: Optional[List[NumType]] = attr.ib(kw_only=True)
-    band_names: Optional[List[str]] = attr.ib(kw_only=True)
-    band_descriptions: Optional[List[str]] = attr.ib(kw_only=True)
+    scales: List[NumType] = attr.ib(kw_only=True)
+    offsets: List[NumType] = attr.ib(kw_only=True)
+    band_names: List[str] = attr.ib(kw_only=True)
+    band_descriptions: List[str] = attr.ib(kw_only=True)
     dataset_statistics: Optional[Sequence[Tuple[float, float]]] = attr.ib(
         default=None, kw_only=True
     )
@@ -495,8 +498,8 @@ class ImageData:
                 if img.height == max_h and img.width == max_w:
                     continue
                 arr = numpy.ma.MaskedArray(
-                    resize_array(img.array.data, max_h, max_w),
-                    mask=resize_array(img.array.mask * 1, max_h, max_w).astype("bool"),
+                    resize_array(img.array.data, max_h, max_w),  # type: ignore [arg-type]
+                    mask=resize_array(img.array.mask * 1, max_h, max_w).astype("bool"),  # type: ignore [arg-type]
                 )
                 img.array = arr
 
@@ -708,8 +711,8 @@ class ImageData:
         resampling_method: RIOResampling = "nearest",
     ) -> "ImageData":
         """Resize data and mask."""
-        data = resize_array(self.array.data, height, width, resampling_method)
-        mask = resize_array(self.array.mask * 1, height, width, resampling_method).astype(
+        data = resize_array(self.array.data, height, width, resampling_method)  # type: ignore [arg-type]
+        mask = resize_array(self.array.mask * 1, height, width, resampling_method).astype(  # type: ignore [arg-type]
             "bool"
         )
         alpha_mask = self.alpha_mask

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/mosaic/backend.py

@@ -14,7 +14,7 @@ from rasterio.features import geometry_mask
 from rio_tiler.constants import WEB_MERCATOR_TMS, WGS84_CRS
 from rio_tiler.errors import NoAssetFoundError, PointOutsideBounds
 from rio_tiler.io import BaseReader, MultiBandReader, MultiBaseReader, Reader
-from rio_tiler.models import ImageData, PointData
+from rio_tiler.models import BandStatistics, ImageData, PointData
 from rio_tiler.mosaic import mosaic_reader
 from rio_tiler.tasks import multi_values_list
 from rio_tiler.types import BBox
@@ -233,10 +233,18 @@ class BaseBackend(BaseReader):
     ############################################################################
     # Not Implemented methods
     # BaseReader required those method to be implemented
-    def statistics(self):
+    def statistics(
+        self,
+        search_options: dict | None = None,
+        **kwargs: Any,
+    ) -> dict[str, BandStatistics]:
         """PlaceHolder for statistics."""
         raise NotImplementedError
 
-    def preview(self):
+    def preview(  # type: ignore
+        self,
+        search_options: dict | None = None,
+        **kwargs: Any,
+    ) -> tuple[ImageData, list[str]]:
         """PlaceHolder for preview."""
         raise NotImplementedError

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/reader.py

@@ -3,7 +3,7 @@
 import contextlib
 import math
 import warnings
-from typing import Callable, Dict, Optional, Tuple, TypedDict, Union
+from typing import Callable, Dict, Optional, Tuple, TypedDict, Union, cast
 
 import numpy
 from affine import Affine
@@ -258,7 +258,11 @@ def read(
         scales = numpy.array(dataset.scales)[numpy.array(indexes) - 1]
         offsets = numpy.array(dataset.offsets)[numpy.array(indexes) - 1]
         if unscale:
-            data = data.astype("float32", casting="unsafe")
+            data = cast(
+                numpy.ma.MaskedArray,
+                data.astype("float32", casting="unsafe"),
+            )
+
             numpy.multiply(data, scales.reshape((-1, 1, 1)), out=data, casting="unsafe")
             numpy.add(data, offsets.reshape((-1, 1, 1)), out=data, casting="unsafe")
 

{rio_tiler-8.0.1 → rio_tiler-8.0.3}/rio_tiler/utils.py

@@ -139,7 +139,7 @@ def get_array_statistics(
     percentiles = percentiles or [2, 98]
 
     if len(data.shape) < 3:
-        data = numpy.ma.expand_dims(data, axis=0)
+        data = numpy.ma.expand_dims(data, axis=0)  # type: ignore [assignment]
 
     output: List[Dict[Any, Any]] = []
     percentiles_names = [f"percentile_{int(p)}" for p in percentiles]