Raise error when Identity has no fillvalue provided and there are nan…

…s in data
MeteoSwiss · Sep 13, 2024 · 5e15e2f · 5e15e2f
1 parent 4d61174
commit 5e15e2f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mlpp_lib/normalizers.py b/mlpp_lib/normalizers.py
@@ -209,6 +209,9 @@ def f(ds: xr.Dataset) -> xr.Dataset:
             ds = ds[self.identity_vars].copy()
             if self.fillvalue:
                 ds = ds.fillna(self.fillvalue)
+            else:
+                if ds.isnull().any():
+                    raise ValueError("Missing values found in the data. Please proivde a fill value.")
             return ds.astype("float32")
 
         return tuple(f(ds) for ds in datasets)