return { b_name: int(res) + 1 for b_name, res in zip(ball_name_list, pred_result_list) }
return {
b_name: int(res) + 1 for b_name, res in zip(ball_name_list, pred_result_list)
}