Bug fixing: batch predictions getter was missing