Update to dataset_driver for testing and inclusion of missing yml files.