Changes in docs for batch predictions and bug fix for test_source, test_dataset