diff --git a/README.md b/README.md index d72b294474ca75f225084e726594ec10036bc30a..5447959277c8482bea23cc84f19fe5f289886cd9 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ roi = ndvi[20:586, 9:572] # Pipeline execution # The actual computation happens here ! -roi.write("output.tif", pixel_type="float") +roi.write("output.tif", "float") ``` pyotb's objects also enable easy interoperability with diff --git a/pyotb/core.py b/pyotb/core.py index dfe2598067fd6ba93050ca7d3555674a25d6de3a..06da0c734b4895f2e58e3c4d7bd1c5ba1b49a27e 100644 --- a/pyotb/core.py +++ b/pyotb/core.py @@ -729,9 +729,9 @@ class App(OTBObject): def write( self, path: str | Path | dict[str, str] = None, - ext_fname: str = "", pixel_type: dict[str, str] | str = None, preserve_dtype: bool = False, + ext_fname: str = "", **kwargs, ) -> bool: """Set output pixel type and write the output raster files. diff --git a/tests/test_core.py b/tests/test_core.py index f766c58ea8a219e3dfd31efd4a731e4da52ede32..99b21deaa78184e7d50869599fdc08bcdbdc3965 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -246,8 +246,8 @@ def test_ndvi_comparison(): ndvi_bandmath = (INPUT[:, :, -1] - INPUT[:, :, [0]]) / (INPUT[:, :, -1] + INPUT[:, :, 0]) ndvi_indices = pyotb.RadiometricIndices(INPUT, {"list": ["Vegetation:NDVI"], "channels.red": 1, "channels.nir": 4}) assert ndvi_bandmath.exp == "((im1b4 - im1b1) / (im1b4 + im1b1))" - assert ndvi_bandmath.write("/tmp/ndvi_bandmath.tif", pixel_type="float") - assert ndvi_indices.write("/tmp/ndvi_indices.tif", pixel_type="float") + assert ndvi_bandmath.write("/tmp/ndvi_bandmath.tif", "float") + assert ndvi_indices.write("/tmp/ndvi_indices.tif", "float") compared = pyotb.CompareImages({"ref.in": ndvi_indices, "meas.in": "/tmp/ndvi_bandmath.tif"}) assert (compared["count"], compared["mse"]) == (0, 0)