Put missing values to 0

WorldCereal · Aug 21, 2024 · f1573cb · f1573cb
1 parent dfd490f
commit f1573cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/presto/inference.py b/presto/inference.py
@@ -112,7 +112,7 @@ def _extract_eo_data(cls, inarr: xr.DataArray) -> Tuple[np.ndarray, np.ndarray]:
                 )
                 idx_valid = values != cls._NODATAVALUE
                 values = cls._preprocess_band_values(values, presto_band)
-                eo_data[:, :, BANDS.index(presto_band)] = values
+                eo_data[:, :, BANDS.index(presto_band)] = values * idx_valid
                 mask[:, :, IDX_TO_BAND_GROUPS[presto_band]] += ~idx_valid
             else:
                 logger.warning(f"Band {org_band} not found in input data.")