use precip_models_cascade if present

pySTEPS · Oct 1, 2024 · c5718cb · c5718cb
1 parent 3d1366b
commit c5718cb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pysteps/blending/steps.py b/pysteps/blending/steps.py
@@ -646,6 +646,10 @@ def forecast(
         # only nans or infs. If so, fill it with the zero value.
         if np.any(precip_cascade[np.isfinite(precip_cascade)]):
             precip_cascade[~np.isfinite(precip_cascade)] = np.nanmin(precip_cascade)
+        elif precip_models_cascade is not None:
+            precip_cascade[~np.isfinite(precip_cascade)] = np.nanmin(
+                precip_models_cascade
+            )
         else:
             done = False
             for t in timesteps: