Handle CMB surface precip.

simonpf · Mar 28, 2024 · 04a0303 · 04a0303
1 parent 89f2762
commit 04a0303
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/gprof_nn/data/training_data.py b/gprof_nn/data/training_data.py
@@ -1006,7 +1006,7 @@ def load_training_data_3d_xtrack_sim(
 
         data = scene[target].data.astype("float32")
 
-        if target in ["surface_precip", "convective_precip"]:
+        if "angles" in scene[target].dims:
             data = interpolate(data, weights)
 
         data = torch.tensor(data)

diff --git a/gprof_nn/data/utils.py b/gprof_nn/data/utils.py
@@ -306,6 +306,7 @@ def save_scene(
             "cloud_water_path",
             "rain_water_path",
             "surface_precip",
+            "surface_precip_combined",
             "convective_precip"
     ]:
         encoding[var] = {"dtype": "float32", "zlib": True}