def get_dataset(dataset_name):
    dataset = None
    if (dataset_name == 'sbpd'):
        dataset = StanfordBuildingParserDataset(dataset_name)
    else:
        logging.fatal('Not one of sbpd')
    return dataset
def get_dataset(dataset_name):
    dataset = None
    if (dataset_name == 'sbpd'):
        dataset = StanfordBuildingParserDataset(dataset_name)
    else:
        logging.fatal('Not one of sbpd')
    return dataset