Skip empty masks

opendatacube · Dec 21, 2023 · ab95b8c · ab95b8c
1 parent 8e52514
commit ab95b8c
Showing 1 changed file with 11 additions and 14 deletions.
diff --git a/eodatasets3/wagl.py b/eodatasets3/wagl.py
@@ -247,20 +247,17 @@ def mask_h5_vector(
     """
     data_array = dataset[:] if hasattr(dataset, "chunks") else dataset
     # Open the gml file
-    try:
-        with fiona.open(mask_string_path) as gml:
-            shapes = [feature["geometry"] for feature in gml]
-            mask_array = rasterio.features.rasterize(
-                shapes,
-                out_shape=data_array.shape,
-                fill=0,
-                transform=Affine.from_gdal(*dataset.attrs["geotransform"]),
-            )
-            return numpy.where(
-                mask_array == 0, data_array, dataset.attrs["no_data_value"]
-            )
-    except ValueError:
-        return data_array
+    with fiona.open(mask_string_path) as gml:
+        shapes = [feature["geometry"] for feature in gml]
+        if not shapes:
+            return data_array
+        mask_array = rasterio.features.rasterize(
+            shapes,
+            out_shape=data_array.shape,
+            fill=0,
+            transform=Affine.from_gdal(*dataset.attrs["geotransform"]),
+        )
+        return numpy.where(mask_array == 0, data_array, dataset.attrs["no_data_value"])
 
 
 def mask_h5_raster(