Add option to create directory in _get_dataset_dir