diff --git a/python/mosaic/api/raster.py b/python/mosaic/api/raster.py index a53b7988b..b0eeb0a99 100644 --- a/python/mosaic/api/raster.py +++ b/python/mosaic/api/raster.py @@ -68,7 +68,7 @@ "rst_tessellate", "rst_transform", "rst_tooverlappingtiles", - "rst_to_overlapping_tiles", # <- deprecated + "rst_to_overlapping_tiles", # <- deprecated "rst_type", "rst_to_overlapping_tiles", "rst_tryopen", @@ -755,7 +755,9 @@ def rst_numbands(raster_tile: ColumnOrName) -> Column: ) -def rst_pixelcount(raster_tile: ColumnOrName, count_nodata: Any = False, count_all: Any = False) -> Column: +def rst_pixelcount( + raster_tile: ColumnOrName, count_nodata: Any = False, count_all: Any = False +) -> Column: """ Parameters ---------- @@ -776,7 +778,7 @@ def rst_pixelcount(raster_tile: ColumnOrName, count_nodata: Any = False, count_a count_nodata = lit(count_nodata) if type(count_all) == bool: - count_all = lit(count_all) + count_all = lit(count_all) return config.mosaic_context.invoke_function( "rst_pixelcount", @@ -1366,10 +1368,10 @@ def rst_tessellate(raster_tile: ColumnOrName, resolution: ColumnOrName) -> Colum def rst_tooverlappingtiles( - raster_tile: ColumnOrName, - width: ColumnOrName, - height: ColumnOrName, - overlap: ColumnOrName, + raster_tile: ColumnOrName, + width: ColumnOrName, + height: ColumnOrName, + overlap: ColumnOrName, ) -> Column: """ Tiles the raster into tiles of the given size. @@ -1388,12 +1390,11 @@ def rst_tooverlappingtiles( def rst_to_overlapping_tiles( - raster_tile: ColumnOrName, - width: ColumnOrName, - height: ColumnOrName, - overlap: ColumnOrName, - ) -> Column: - + raster_tile: ColumnOrName, + width: ColumnOrName, + height: ColumnOrName, + overlap: ColumnOrName, +) -> Column: return rst_tooverlappingtiles(raster_tile, width, height, overlap) diff --git a/python/test/test_raster_functions.py b/python/test/test_raster_functions.py index 68288be69..a5f27fed9 100644 --- a/python/test/test_raster_functions.py +++ b/python/test/test_raster_functions.py @@ -59,7 +59,7 @@ def test_raster_scalar_functions(self): .withColumn( "rst_mapalgebra", api.rst_mapalgebra( - array("tile", "rst_initnodata"), + array("tile_from_file", "rst_initnodata"), lit('{"calc": "A+B", "A_index": 0, "B_index": 1}'), ), )