support limiting the columns used for batching