changes to yamlmodelrunner for new dataset style