You should be able to pass a list of column names into `Dataset.to_batches` https://arrow.apache.org/docs/python/generated/pyarrow.dataset.Dataset.html#pyarrow.dataset.Dataset.to_batches
You should be able to pass a list of column names into
Dataset.to_batcheshttps://arrow.apache.org/docs/python/generated/pyarrow.dataset.Dataset.html#pyarrow.dataset.Dataset.to_batches