return early if no data

sentinel-hub · Jun 12, 2024 · 33b66a6 · 33b66a6
1 parent efa49d4
commit 33b66a6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/eolearn/mask/snow_mask.py b/eolearn/mask/snow_mask.py
@@ -66,6 +66,10 @@ def _apply_dilation(self, snow_masks: np.ndarray) -> np.ndarray:
 
     def execute(self, eopatch: EOPatch) -> EOPatch:
         bands = eopatch[self.bands_feature][..., self.band_indices]
+        if bands.shape[0] == 0:
+            eopatch[self.mask_feature] = np.zeros((*bands.shape[:-1], 1), dtype=bool)
+            return eopatch
+
         with np.errstate(divide="ignore", invalid="ignore"):
             # (B03 - B11) / (B03 + B11)
             ndsi = (bands[..., 0] - bands[..., 3]) / (bands[..., 0] + bands[..., 3])