Fix incorrect mention to callbacks in Y.batch API docs

This was a left-over from the previous implementation