Consider the data mask when creating raster from numpy input

noaa-ocs-modeling · Aug 23, 2023 · 2b7e84e · 2b7e84e
1 parent ff56f1a
commit 2b7e84e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ocsmesh/utils.py b/ocsmesh/utils.py
@@ -2157,6 +2157,8 @@ def raster_from_numpy(
         crs=crs,
         transform=transform,
     ) as dst:
+        if isinstance(data, np.ma.MaskedArray):
+            dst.nodata = data.fill_value
         dst.write(data, 1)