class LazyBatch(LazyDict):
    def format(self, key):
        return self.formatter.format_column(self.pa_table.select([key]))
/Users/nielsrogge/Documents/python_projecten/datasets/src/datasets/formatting/formatting.py