diff --git a/odc/geo/_dask.py b/odc/geo/_dask.py index a8d7d4e..68ca299 100644 --- a/odc/geo/_dask.py +++ b/odc/geo/_dask.py @@ -70,7 +70,7 @@ def _do_chunked_reproject( return dst -def _dask_rio_reproject( +def dask_rio_reproject( src: da.Array, s_gbox: Union[GeoBox, GCPGeoBox], d_gbox: GeoBox, @@ -94,7 +94,6 @@ def with_yx(a, yx): name: str = kwargs.pop("name", "reproject") - assert isinstance(s_gbox, GeoBox) gbt_src = GeoboxTiles(s_gbox, src.chunks[ydim : ydim + 2]) gbt_dst = GeoboxTiles(d_gbox, chunks) d2s_idx = gbt_dst.grid_intersect(gbt_src) diff --git a/odc/geo/_xr_interop.py b/odc/geo/_xr_interop.py index b149eca..a1637c3 100644 --- a/odc/geo/_xr_interop.py +++ b/odc/geo/_xr_interop.py @@ -821,9 +821,9 @@ def _xr_reproject_da( dst_nodata = src_nodata if is_dask_collection(src): - from ._dask import _dask_rio_reproject + from ._dask import dask_rio_reproject - dst: Any = _dask_rio_reproject( + dst: Any = dask_rio_reproject( src.data, src_gbox, dst_geobox,