The only addition is the else (missing forecast period).
The grid_set was already there, so i'll just add a test for missing forecast period as time is short.