fixing small bug, removing the restriction to have a full batch finished before we can predict