Modified training env setup scripts