skip computation of next forcings if current step is the last one (#71)

ecmwf · Dec 5, 2024 · b1d2560 · b1d2560
1 parent 62fff47
commit b1d2560
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/anemoi/inference/runner.py b/src/anemoi/inference/runner.py
@@ -296,6 +296,10 @@ def forecast(self, lead_time, input_tensor_numpy, input_state):
 
             yield result
 
+            # No need to prepare next input tensor if we are at the last step
+            if s == steps - 1:
+                continue
+
             # Update  tensor for next iteration
 
             check[:] = reset