Definition of batch function in the Data module line
|> Stream.chunk_every(batch_size, :discard)
should be
|> Stream.chunk_every(batch_size, batch_size, :discard)
Definition of batch function in the Data module line
|> Stream.chunk_every(batch_size, :discard)
should be
|> Stream.chunk_every(batch_size, batch_size, :discard)