Adjustments required as part of ECMWF ML PP WP1 Task 1.3

CHANGELOG.md

@@ -12,6 +12,7 @@ Keep it human-readable, your future self will thank you!
 ### Added
 
 ### Changed
+- use `PS` if `lsm` is not available as prognostic template
 
 ### Removed
 

src/anemoi/inference/checkpoint/metadata/__init__.py

@@ -1,10 +1,10 @@
 # (C) Copyright 2023 European Centre for Medium-Range Weather Forecasts.
+# (C) Copyright 2024 Deutscher Wetterdienst.
 # This software is licensed under the terms of the Apache Licence Version 2.0
 # which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
-# In applying this licence, ECMWF does not waive the privileges and immunities
-# granted to it by virtue of its status as an intergovernmental organisation
-# nor does it submit to any jurisdiction.
-
+# In applying this licence, the above institution do not waive the privileges
+# and immunities granted to it by virtue of its status as an intergovernmental
+# organisation  nor does it submit to any jurisdiction.
 
 import json
 import logging
@@ -115,6 +115,22 @@ def hour_steps(self):
 
         return int(n[:-1]) * {"h": 1, "d": 24}[n[-1]]
 
+    @cached_property
+    def max_refinement_level_c(self):
+        """Returns refinement level of ICON input dataset, in [0,1,2,3,4,5,6,7].
+        If not defined it returns None.
+        """
+
+        return (
+            self._metadata.get("dataset", {})
+            .get("specific", {})
+            .get("forward", {})
+            .get("forward", {})
+            .get("attrs", {})
+            .get("data_request", {})
+            .get("max_refinement_level_c")
+        )
+
     ###########################################################################
     # Indices
 

src/anemoi/inference/runner.py

@@ -1,9 +1,10 @@
 # (C) Copyright 2024 European Centre for Medium-Range Weather Forecasts.
+# (C) Copyright 2024 Deutscher Wetterdienst.
 # This software is licensed under the terms of the Apache Licence Version 2.0
 # which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
-# In applying this licence, ECMWF does not waive the privileges and immunities
-# granted to it by virtue of its status as an intergovernmental organisation
-# nor does it submit to any jurisdiction.
+# In applying this licence, the above institution do not waive the privileges
+# and immunities granted to it by virtue of its status as an intergovernmental
+# organisation  nor does it submit to any jurisdiction.
 
 
 import datetime
@@ -312,7 +313,10 @@ def get_most_recent_datetime(input_fields):
 
         most_recent_datetime = get_most_recent_datetime(input_fields)
         reference_fields = [f for f in input_fields if f.datetime()["valid_time"] == most_recent_datetime]
-        prognostic_template = reference_fields[self.checkpoint.variable_to_index["lsm"]]
+        if "lsm" in self.checkpoint.variable_to_index:
+            prognostic_template = reference_fields[self.checkpoint.variable_to_index["lsm"]]
+        else:
+            prognostic_template = reference_fields[self.checkpoint.variable_to_index["PS"]]
 
         accumulated_output = np.zeros(
             shape=(len(diagnostic_output_mask), number_of_grid_points),