Improving yield predictions in drought-prone environments through trait phenotyping and advanced modelling