fix(netcdf): Fix context time attributes (#204)

Author: gmertes

src/anemoi/inference/outputs/netcdf.py

@@ -91,10 +91,12 @@ def open(self, state: State) -> None:
 
         time = 0
         self.reference_date = state["date"]
-        if hasattr(self.context, "time_step") and hasattr(self.context, "lead_time"):
-            time = self.context.lead_time // self.context.time_step
-        if hasattr(self.context, "reference_date"):
-            self.reference_date = self.context.reference_date
+        if (time_step := getattr(self.context, "time_step", None)) and (
+            lead_time := getattr(self.context, "lead_time", None)
+        ):
+            time = lead_time // time_step
+        if reference_date := getattr(self.context, "reference_date", None):
+            self.reference_date = reference_date
 
         with LOCK:
             self.values_dim = self.ncfile.createDimension("values", values)

src/anemoi/inference/runner.py

@@ -214,10 +214,6 @@ def run(
 
         lead_time = to_timedelta(lead_time)
 
-        # This may be used but Output objects to compute the step
-        self.lead_time = lead_time
-        self.time_step = self.checkpoint.timestep
-
         with ProfilingRunner(self.use_profiler):
             with ProfilingLabel("Prepare input tensor", self.use_profiler):
                 input_tensor = self.prepare_input_tensor(input_state)

src/anemoi/inference/runners/default.py

@@ -15,6 +15,7 @@
 from typing import List
 
 from anemoi.utils.config import DotDict
+from anemoi.utils.dates import frequency_to_timedelta as to_timedelta
 from pydantic import BaseModel
 
 from anemoi.inference.config import Configuration
@@ -86,6 +87,12 @@ def execute(self) -> None:
         if self.config.description is not None:
             LOG.info("%s", self.config.description)
 
+        lead_time = to_timedelta(self.config.lead_time)
+
+        # This may be used by Output objects to compute the step
+        self.lead_time = lead_time
+        self.time_step = self.checkpoint.timestep
+
         input = self.create_input()
         output = self.create_output()
 
@@ -104,7 +111,7 @@ def execute(self) -> None:
         output.open(state)
         output.write_initial_state(state)
 
-        for state in self.run(input_state=input_state, lead_time=self.config.lead_time):
+        for state in self.run(input_state=input_state, lead_time=lead_time):
             for processor in post_processors:
                 LOG.info("Post processor: %s", processor)
                 state = processor.process(state)