fix: Dask reproject from GCP based source

- remove incorrect type assert - make dask_rio_reproject "public"
opendatacube · Jun 11, 2024 · b6b9ce3 · b6b9ce3
1 parent ce04120
commit b6b9ce3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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,