Fix batch_size sanity check logic for `split_batches ` (#2344)
* fix
* lets raise an error
* Update error message
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
* fix error message style
---------
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>