Skip to content

Commit efeef39

Browse files
committed
make mypy happy
1 parent 0979694 commit efeef39

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

xvec/zonal.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -495,7 +495,7 @@ def _zonal_stats_exactextract(
495495

496496
def _variable_zonal_exactextract(
497497
acc,
498-
geometry: Sequence[shapely.Geometry] | xr.DataArray,
498+
geometry: xr.DataArray,
499499
x_coords: Hashable,
500500
y_coords: Hashable,
501501
stats: str | Callable | Sequence[str | Callable | tuple] = "mean",
@@ -612,8 +612,8 @@ def _variable_zonal_exactextract(
612612
),
613613
)
614614

615-
if original_is_ds:
616-
result = result.to_dataset("variable")
615+
if original_is_ds and isinstance(result, xr.DataArray):
616+
result = result.to_dataset("variable") # type: ignore
617617
acc._obj = original_ds
618618

619619
result.attrs = acc._obj.attrs
@@ -633,7 +633,7 @@ def _agg_exactextract(
633633
x_coords: Hashable,
634634
y_coords: Hashable,
635635
stats: str | Callable | Iterable[str | Callable | tuple] = "mean",
636-
name: str = "geometry",
636+
name: Hashable = "geometry",
637637
original_is_ds: bool = False,
638638
strategy: Literal["feature-sequential", "raster-sequential"] = "raster-sequential",
639639
nodata: Any = None,

0 commit comments

Comments
 (0)