diff --git a/components/sud_forecast.py b/components/sud_forecast.py index cd8238d..bd6bd27 100644 --- a/components/sud_forecast.py +++ b/components/sud_forecast.py @@ -156,8 +156,8 @@ class SudForecastEstimator: # re-assigns self.step to retrigger it) - only the *first* call # for a given index is its actual start. step_starts.setdefault(sud.index, t[-1]) - pot = step.get('pot', {}) - params = sud.derive_plant_params(pot.get('grain_mass', 0), pot.get('water_mass', 0)) + step_pot = step.get('pot', {}) + params = sud.derive_plant_params(step_pot.get('grain_mass', 0), step_pot.get('water_mass', 0)) pot.set_plant_params(params) tc.set_model_plant_params(params) if sud.state == SudState.RAMPING and step['temperature'] is not None: