getting rid of assert regarding integral batch sizes