Fix for time dependent models, in case there are missing training instances for certain steps (e.g. LWR trajectory)