C3W1_UNQ_C9 T learner predict method returned unexpected type

I am returning the pred_risk_reduction in the T learner prediction function as an np.array but I get an error later when I run the quantile_benefit() cell. Can someone point out to me what I am doing wrong? Thanks

please avoid posting grade function codes on public post threads as they grade your assignment and is considered against community guidelines.

here is post comment which will help you debug your codes

This is helpful. Thank you