Merge branch 'develop' into feature/initial-state-output

CHANGELOG.md

@@ -30,6 +30,7 @@ Keep it human-readable, your future self will thank you!
 - Update copyright notice
 - Fix `__version__` import in init
 - use earthkit-data 0.11.2
+- Fix SimpleRunner
 
 ### Removed
 

src/anemoi/inference/runner.py

@@ -145,6 +145,9 @@ def add_initial_forcings_to_input_state(self, input_state):
         # TODO: Check for user provided forcings
 
         for source in self.constant_forcings_inputs:
+            if source is None:
+                # When the constants are already in the input state
+                continue
             LOG.info("Constant forcings input: %s %s (%s)", source, source.variables, dates)
             arrays = source.load_forcings(input_state, dates)
             for name, forcing in zip(source.variables, arrays):

src/anemoi/inference/runners/simple.py

@@ -10,15 +10,46 @@
 
 import logging
 
+from ..forcings import ComputedForcings
+from ..forcings import Forcings
 from ..runner import Runner
 from . import runner_registry
 
 LOG = logging.getLogger(__name__)
 
+# This is because forcings are assumed to already be in the
+# state dictionary, so we don't need to load them from anywhere.
+
+
+class NoForcings(Forcings):
+    """No forcings."""
+
+    def __init__(self, context, variables, mask):
+        super().__init__(context)
+        self.variables = variables
+        self.mask = mask
+        self.kinds = dict(unknown=True)
+
+    def load_forcings(self, state, date):
+        pass
+
 
 @runner_registry.register("simple")
 class SimpleRunner(Runner):
     """Use that runner when using the low level API."""
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+
+    def create_constant_computed_forcings(self, variables, mask):
+        result = ComputedForcings(self, variables, mask)
+        LOG.info("Constant computed forcing: %s", result)
+        return result
+
+    def create_dynamic_computed_forcings(self, variables, mask):
+        result = ComputedForcings(self, variables, mask)
+        LOG.info("Dynamic computed forcing: %s", result)
+        return result
+
+    def create_constant_coupled_forcings(self, variables, mask):
+        return None