bart_pred_for_test_data fix