# build dataset according to given 'dataset_file'
def build_dataset(args):
    dataset_file = args.dataset_file.upper()
    if dataset_file == 'SHHA':
        from crowd_datasets.SHHA.loading_data import loading_data
        return loading_data

    raise ValueError(f"Unsupported dataset_file: {args.dataset_file}. Use SHHA.")