uber / causalml

Uplift modeling and causal inference with machine learning algorithms
Other
4.87k stars 756 forks source link

return_ci=True is not properly passed in get_ate_ci of the Sensitivity class in sensitivity.py #757

Closed lee-junseok closed 2 months ago

lee-junseok commented 2 months ago

Describe the bug Some learner's estimate_ate function requires return_ci = True to return ate_lower, ate_upper together with ate. get_ate_ci of the Sensitivity class does not pass return_ci = True to estimate_ate.

To Reproduce Steps to reproduce the behavior: Run sensitivity_analysis method with a class object Sensitivity(..., learner = XGBTRegressor(...), ...)

It gives ValueError: not enough values to unpack (expected 3, got 1)

Expected behavior Get sensitivity analysis result without a ValueError

Screenshots

image

Learner: XGBTRegressor()

Environment (please complete the following information):

Additional context None