Parameterize StreamTx with the batch size