Question: how to do sensitivity analysis with S_learner

[huanyi89]

I am using s_learner for my causal modeling (BaseSClassifier(learner=XGBClassifier))
However, when I trying to access the sensitivity for the model, it reported an error for the function
ate_new, ate_new_lower, ate_new_upper = self.get_ate_ci(X, p, treatment_new, y)
ValueError: not enough values to unpack (expected 3, got 1)

My main question here is can I do the sensitivity analysis for S_learner? If yes, how to do that. I am not sure if I missed some settings that can make it work.

Thank you