Bug fixing: resume crashed checking # of predictions

- The checkpoint function did not return a tuple