File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -495,7 +495,7 @@ def _zonal_stats_exactextract(
495495
496496def _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 ,
You can’t perform that action at this time.
0 commit comments