def a_single_column_apply_fn(batch, columns):
    """Simple function that applies to a single column."""
    assert len(columns) == 1 and type(batch[columns[0]][0]) == int
    return [e * 7 + 3.14 for e in batch[columns[0]]]